Update app.py

app.py

@@ -1,64 +1,30 @@
 import gradio as gr
 from transformers import pipeline
-import logging
-
-# إعداد logging (يشبه terminal logs)
-logging.basicConfig(level=logging.INFO)
-
-logging.info("Starting Flogistic AI...")
 
 # تحميل نموذج خفيف
-generator = pipeline(
-    "text-generation",
-    model="distilgpt2"
-)
-
-# ذاكرة محادثة
-chat_history = []
-
-def generate_response(user_input):
-    global chat_history
-
-    logging.info(f"User Input: {user_input}")
-
-    prompt = ""
-    for u, a in chat_history:
-        prompt += f"User: {u}\nAI: {a}\n"
+generator = pipeline("text-generation", model="distilgpt2")
 
-    prompt += f"User: {user_input}\nAI:"
+# الرد
+def respond(message, chat_history):
+    if chat_history is None:
+        chat_history = []
 
-    result = generator(prompt, max_length=150, do_sample=True)
+    response = generator(
+        message,
+        max_new_tokens=60,
+        do_sample=True
+    )[0]["generated_text"]
 
-    response = result[0]["generated_text"].split("AI:")[-1].strip()
+    chat_history.append((message, response))
+    return "", chat_history
 
-    chat_history.append((user_input, response))
-
-    logging.info(f"AI Response: {response}")
-
-    return chat_history
-
-# واجهة احترافية
+# واجهة بسيطة ومستقرة
 with gr.Blocks() as demo:
-    gr.Markdown("# 🚀 Flogistic AI Assistant")
+    gr.Markdown("# 🤖 Flogistic AI Assistant")
 
     chatbot = gr.Chatbot()
     msg = gr.Textbox(placeholder="Type your message...")
 
-    def respond(message, history):
-        history = generate_response(message)
-        return "", history
-
     msg.submit(respond, [msg, chatbot], [msg, chatbot])
 
-demo.launch()
-log_output = gr.Textbox(label="System Logs")
-
-def respond_with_logs(message, history):
-    history = generate_response(message)
-    logs = f"Last input: {message}\nHistory length: {len(history)}"
-    return "", history, logs
-msg.submit(
-    respond_with_logs,
-    [msg, chatbot],
-    [msg, chatbot, log_output]
-)
+demo.launch()