Update app.py

app.py

@@ -1,37 +1,59 @@
-import streamlit as st
+# gradio_g4f_chat.py
+import gradio as gr
 from g4f.client import Client
 
+# --- Инициализация ---
 client = Client()
-
-st.set_page_config(
-    page_title="GPT-4o Chat",
-    page_icon="📝",
-    layout="centered"
+chat_history = []  # можно использовать st.session_state аналог — через closure или state
+
+# --- Настройки ---
+SYSTEM_PROMPT = "You are a helpful assistant."
+
+def respond(message, history):
+    global chat_history
+    
+    # Формируем полный контекст
+    messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+    for human, assistant in history:
+        messages.append({"role": "user", "content": human})
+        messages.append({"role": "assistant", "content": assistant})
+    messages.append({"role": "user", "content": message})
+
+    try:
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=messages
+        )
+        bot_message = response.choices[0].message.content
+    except Exception as e:
+        bot_message = f"Ошибка связи с ИИ: {str(e)}"
+
+    # Добавляем в историю
+    chat_history.append((message, bot_message))
+    
+    return bot_message
+
+# --- Создание интерфейса ---
+demo = gr.ChatInterface(
+    fn=respond,
+    title="📝 GPT-4o-mini Chat via g4f",
+    description="Общайся с GPT-4o-mini без API-ключа. Powered by g4f.",
+    examples=[
+        "Привет, кто ты?",
+        "Расскажи про искусственный интеллект",
+        "Напиши стих про робота"
+    ],
+    retry_btn="🔄 Повторить",
+    undo_btn="↩️ Отменить",
+    clear_btn="🗑️ Очистить"
 )
 
-if "chat_history" not in st.session_state:
-    st.session_state.chat_history = []
-
-st.title("GPT-4o-mini Chat 🤖")
-
-for message in st.session_state.chat_history:
-    with st.chat_message(message['role']):
-        st.markdown(message['content'])
-
-user_prompt = st.chat_input("Ask GPT-4o-mini")
-
-if user_prompt:
-    st.chat_message("user").markdown(user_prompt)
-    st.session_state.chat_history.append({"role": "user", "content": user_prompt})
-
-    response = client.chat.completions.create(
-        model="gpt-4o-mini",
-        messages=[
-            {"role": "system", "content": "You are a helpful assistant"},
-            *st.session_state.chat_history
-        ]
-    )
-    final_response = response.choices[0].message.content
-    st.session_state.chat_history.append({"role": "assistant", "content": final_response})
-    with st.chat_message("assistant"):
-        st.markdown(final_response)
+# --- Запуск ---
+if __name__ == "__main__":
+    demo.queue()  # для асинхронной обработки
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False,  # если хочешь публичную ссылку — поставь True
+        debug=True
+    )