Update app.py

app.py

@@ -1,52 +1,43 @@
 import streamlit as st
-import g4f
+from g4f import ChatCompletion
 
-# Set the title of the app
-st.title("Chat with GPT-4 Free")
+# List of available models
+models = [
+    "gpt-4o", "gpt-4o-mini", "gpt-4",
+    "gpt-4-turbo", "gpt-3.5-turbo",
+    "claude-3.5-sonnet", "claude-3-opus", "claude-3-haiku"
+]
 
-# Initialize chat history in session state
+# Initialize chat history
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-# Model selection
-model_options = {
-    "GPT-4 Optimized": g4f.models.gpt_4o,
-    "GPT-4": g4f.models.gpt_4,
-    "GPT-4 Turbo": g4f.models.gpt_4_turbo,
-    "GPT-4 Optimized Mini": g4f.models.gpt_4o_mini
-}
-
-selected_model = st.selectbox("Choose a model:", list(model_options.keys()))
+# Streamlit app title
+st.title("Chat with AI Models")
 
-# Function to generate responses from g4f
-def generate_response(prompt):
-    response = g4f.ChatCompletion.create(
-        model=model_options[selected_model],
-        messages=[{"role": "user", "content": prompt}],
-        stream=True  # Enable streaming for real-time response
-    )
-    return response
+# Model selection
+selected_model = st.selectbox("Choose a model:", models)
 
 # Display chat messages from history
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 
-# Accept user input
-if prompt := st.chat_input("Type your message here..."):
-    # Display user message in chat message container
-    with st.chat_message("user"):
-        st.markdown(prompt)
-    
-    # Add user message to chat history
-    st.session_state.messages.append({"role": "user", "content": prompt})
+# User input
+if user_input := st.chat_input("What do you want to ask?"):
+    # Display user message
+    st.chat_message("user").markdown(user_input)
+    st.session_state.messages.append({"role": "user", "content": user_input})
 
-    # Generate and display AI response
+    # Get response from selected model
+    response = ChatCompletion.create(
+        model=selected_model,
+        messages=st.session_state.messages
+    )
+    
+    # Display assistant response
     with st.chat_message("assistant"):
-        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": msg})
+        st.markdown(response['choices'][0]['message']['content'])
+    
+    # Add assistant response to chat history
+    st.session_state.messages.append({"role": "assistant", "content": response['choices'][0]['message']['content']})