Update app.py

app.py

@@ -1,120 +1,33 @@
-import streamlit as st
-from streamlit_chat import message
-from transformers import pipeline
-import base64
-
-# تنظیمات اولیه صفحه
-st.set_page_config(page_title="دستیار هوش مصنوعی پیشرفته", page_icon="🧠", layout="wide")
-
-# استایل CSS سفارشی
-st.markdown(
-    """
-    <style>
-        @keyframes rotate {
-            0% { transform: rotate(0deg); }
-            100% { transform: rotate(360deg); }
-        }
-        @keyframes float {
-            0%, 100% { transform: translateY(0); }
-            50% { transform: translateY(-10px); }
-        }
-        .brain-icon {
-            display: flex;
-            justify-content: center;
-            align-items: center;
-            margin-bottom: 20px;
-        }
-        .brain-icon img {
-            width: 100px;
-            height: 100px;
-            animation: rotate 10s linear infinite, float 3s ease-in-out infinite;
-        }
-        .main {
-            background-color: #f0f3f5 !important;
-            font-family: 'Vazirmatn', sans-serif;
-        }
-        div[data-testid="stTextArea"] > div {
-            border-radius: 30px;
-            border: 2px solid #3498db;
-            padding: 10px;
-        }
-        div[data-testid="stTextArea"] textarea {
-            font-size: 16px !important;
-            font-family: 'Vazirmatn', sans-serif;
-        }
-        button[data-testid="stButton"] {
-            background-color: #3498db !important;
-            color: white !important;
-            border: none !important;
-            border-radius: 50px !important;
-            font-size: 18px !important;
-            padding: 10px 20px !important;
-        }
-        button[data-testid="stButton"]:hover {
-            background-color: #2980b9 !important;
-        }
-    </style>
-    """,
-    unsafe_allow_html=True,
-)
+"""
+این فایل یک سرور ساده Flask اجرا می‌کند که به فایل index.html متصل است.
+مدل GPT-2 از Hugging Face برای تولید پاسخ‌ها استفاده می‌شود.
+"""
 
-# افزودن فونت فارسی (Vazirmatn)
-st.markdown(
-    """
-    <link href="https://cdn.jsdelivr.net/gh/rastikerdar/vazirmatn@v33.003/Vazirmatn-font-face.css" rel="stylesheet" type="text/css" />
-    """,
-    unsafe_allow_html=True,
-)
-
-# نمایش آیکون مغز با انیمیشن
-st.markdown(
-    """
-    <div class="brain-icon">
-        <img src="https://upload.wikimedia.org/wikipedia/commons/6/6e/Brain_in_a_jar.svg" alt="Brain Icon">
-    </div>
-    """,
-    unsafe_allow_html=True,
-)
+from flask import Flask, request, jsonify, render_template
+from transformers import pipeline
 
-# عنوان صفحه
-st.title("🧠 دستیار هوش مصنوعی پیشرفته")
+# بارگذاری مدل از Hugging Face
+generator = pipeline("text-generation", model="gpt2")
 
-# حالت جلسه برای ذخیره تاریخچه چت
-if "messages" not in st.session_state:
-    st.session_state.messages = []  # ذخیره پیام‌های کاربر و دستیار
+# ایجاد برنامه Flask
+app = Flask(__name__)
 
-# اتصال به مدل هوش مصنوعی (GPT از Hugging Face)
-@st.cache_resource
-def load_model():
-    return pipeline("text-generation", model="gpt2")
+@app.route("/")
+def index():
+    return render_template("index.html")  # فایل index.html
 
-generator = load_model()
+@app.route("/predict", methods=["POST"])
+def predict():
+    data = request.get_json()
+    user_message = data.get("message", "")
+    if not user_message.strip():
+        return jsonify({"reply": "لطفاً یک سوال وارد کنید!"})
 
-# نمایش پیام‌های چت
-for msg in st.session_state.messages:
-    if msg["is_user"]:
-        message(msg["content"], is_user=True)  # پیام کاربر
-    else:
-        message(msg["content"], is_user=False)  # پیام دستیار
+    # تولید پاسخ با مدل
+    response = generator(user_message, max_length=50, num_return_sequences=1)
+    reply = response[0]["generated_text"]
+    return jsonify({"reply": reply})
 
-# ورودی پیام جدید
-with st.container():
-    user_input = st.text_input(
-        "سوال خود را اینجا بنویسید...",
-        key="user_input",
-        placeholder="مثلاً تحلیل قیمت بیت‌کوین...",
-        label_visibility="collapsed",
-    )
-    if st.button("ارسال"):
-        if user_input.strip():  # اگر پیام خالی نباشد
-            # افزودن پیام کاربر به تاریخچه
-            st.session_state.messages.append({"content": user_input, "is_user": True})
-            
-            # دریافت پاسخ از مدل
-            response = generator(user_input, max_length=50, num_return_sequences=1)[0]["generated_text"]
-            
-            # افزودن پاسخ دستیار به تاریخچه
-            st.session_state.messages.append({"content": response, "is_user": False})
-            
-            # به‌روزرسانی صفحه
-            st.experimental_rerun()
+# اجرای برنامه
+if __name__ == "__main__":
+    app.run(debug=True, host="0.0.0.0", port=8000)