Update app.py

app.py

@@ -1,6 +1,4 @@
 import streamlit as st
-import requests
-from bs4 import BeautifulSoup
 import g4f
 
 # Set the title of the app
@@ -10,65 +8,23 @@ st.title("Chat with GPT-4 Free")
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-# Dropdown for model selection
+# Model selection
 model_options = {
-    "GPT-3.5 Turbo": g4f.models.gpt_3.5_turbo,
+    "GPT-4 Optimized": g4f.models.gpt_4o,
     "GPT-4": g4f.models.gpt_4,
-    "GPT-4o": g4f.models.gpt_4o,
+    "GPT-4 Turbo": g4f.models.gpt_4_turbo,
+    "GPT-4 Optimized Mini": g4f.models.gpt_4o_mini
 }
 
-selected_model = st.sidebar.selectbox("Select Model", list(model_options.keys()))
-
-# Toggle for internet search functionality
-search_enabled = st.sidebar.checkbox("Enable Internet Search")
-
-# Function to perform a Google search
-def google_search(query):
-    headers = {
-        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/74.0.3729.169 Safari/537.36'
-    }
-    
-    # Construct the search URL
-    url = f"https://www.google.com/search?q={query}"
-    
-    # Send the request to Google
-    response = requests.get(url, headers=headers)
-    
-    # Parse the HTML content
-    soup = BeautifulSoup(response.text, 'html.parser')
-    
-    # Find all search result containers
-    results = soup.find_all('div', class_='g')
-
-    # Extract titles and summaries
-    search_results = []
-    
-    for result in results:
-        title = result.find('h3')
-        summary = result.find('span', class_='aCOpRe')
-        
-        if title and summary:
-            search_results.append({
-                'title': title.text,
-                'summary': summary.text
-            })
-    
-    return search_results
+selected_model = st.selectbox("Choose a model:", list(model_options.keys()))
 
 # Function to generate responses from g4f
-def generate_response(prompt, model, search=False):
-    if search:
-        # Perform a web search and return results
-        search_results = google_search(prompt)
-        formatted_results = "\n".join([f"{result['title']}: {result['summary']}" for result in search_results])
-        return f"Search Results:\n{formatted_results}" if formatted_results else "No results found."
-    
+def generate_response(prompt):
     response = g4f.ChatCompletion.create(
-        model=model,
+        model=model_options[selected_model],
         messages=[{"role": "user", "content": prompt}],
         stream=True  # Enable streaming for real-time response
     )
-    
     return response
 
 # Display chat messages from history
@@ -87,10 +43,10 @@ if prompt := st.chat_input("Type your message here..."):
 
     # Generate and display AI response
     with st.chat_message("assistant"):
-        response = generate_response(prompt, model_options[selected_model], search=search_enabled)
-        
-        if isinstance(response, str):  # Check if the response is a string
-            st.markdown(response)
+        response = generate_response(prompt)
+        for msg in response:
+            if isinstance(msg, str):  # Check if the message is a string
+                st.markdown(msg)
 
         # Add assistant's response to chat history
-        st.session_state.messages.append({"role": "assistant", "content": response})
+        st.session_state.messages.append({"role": "assistant", "content": msg})