Update app.py

app.py

@@ -1,82 +1,42 @@
 import streamlit as st
-from g4f.client import Client
+import g4f
 
-# Инициализация клиента GPT
-client = Client()
+# Set the title of the app
+st.title("Chat with GPT-4 Free")
 
-# Настройки страницы
-st.set_page_config(page_title="Chat with GPT", page_icon="🤖", layout="wide")
+# Initialize chat history in session state
+if "messages" not in st.session_state:
+    st.session_state.messages = []
 
-# Заголовок и описание
-st.title("Chat with GPT Models")
-st.write("Welcome to the GPT chat interface. Choose a model and start chatting!")
-
-# Выбор модели
-model_options = {
-    "gpt-4": "gpt-4",
-    "gpt-4o": "gpt-4o",
-    "gpt-4o-mini": "gpt-4o-mini",
-    "gpt-3.5-turbo": "gpt-3.5-turbo",
-    "gpt-4-turbo": "gpt-4-turbo",
-    "claude-3.5-sonnet": "claude-3.5-sonnet",
-    "claude-3-opus": "claude-3-opus",
-    "claude-3-haiku": "claude-3-haiku"
-}
-
-selected_model = st.selectbox("Choose a model:", list(model_options.keys()))
-
-# Создание интерфейса для чата
-chat_container = st.container()
-
-# Создание формы для ввода сообщения
-with st.form(key='chat_form'):
-    user_input = st.text_area("You:", height=100)
-    send_button = st.form_submit_button("Send")
-
-# Обработка отправки сообщения
-if send_button and user_input:
-    # Отправка запроса к выбранной модели GPT
-    response = client.chat.completions.create(
-        model=model_options[selected_model],
-        messages=[{"role": "user", "content": user_input}],
+# Function to generate responses from g4f
+def generate_response(prompt):
+    response = g4f.ChatCompletion.create(
+        model=g4f.models.gpt_4,
+        messages=[{"role": "user", "content": prompt}],
+        stream=True  # Enable streaming for real-time response
     )
+    return response
+
+# 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)
     
-    # Получение ответа
-    gpt_response = response.choices[0].message.content
-    
-    # Отображение сообщений в чате
-    with chat_container:
-        st.write(f"**You:** {user_input}")
-        st.write(f"**{selected_model}:** {gpt_response}")
-
-# Пример стилизации с использованием CSS
-st.markdown(
-    """
-    <style>
-    .stTextInput > div > div > textarea {
-        font-size: 18px;
-    }
-    .stButton > button {
-        background-color: #4CAF50;
-        color: white;
-        font-size: 16px;
-        padding: 10px 20px;
-        border: none;
-        cursor: pointer;
-    }
-    .stButton > button:hover {
-        background-color: #45a049;
-    }
-    </style>
-    """,
-    unsafe_allow_html=True
-)
-
-# Пример добавления изображения
-st.image("https://via.placeholder.com/150", caption="GPT Models", use_column_width=True)
-
-# Пример добавления разделителя
-st.markdown("---")
-
-# Пример добавления ссылки
-st.markdown("[Learn more about GPT models](https://platform.openai.com/docs/models)")
+    # Add user message to chat history
+    st.session_state.messages.append({"role": "user", "content": prompt})
+
+    # Generate and display AI 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})