Update app.py

app.py

@@ -14,8 +14,6 @@ model = load_model()
 # اپلیکیشن Streamlit
 def main():
     st.set_page_config(page_title="دستیار هوش مصنوعی", layout="wide")
-    
-    # عنوان برنامه
     st.title("دستیار هوش مصنوعی پیشرفته")
 
     # نمایش فایل HTML
@@ -23,22 +21,14 @@ def main():
         html_content = html_file.read()
     st.components.v1.html(html_content, height=800, scrolling=True)
 
-    # مدیریت پیام‌های دریافتی از جاوااسکریپت
-    if "messages" not in st.session_state:
-        st.session_state["messages"] = []
-
-    message = st.text_input("پیام ورودی از جاوااسکریپت:")
-    if message:
-        st.session_state["messages"].append({"user": message})
-        response = model(message, max_length=50, num_return_sequences=1)[0]["generated_text"]
-        st.session_state["messages"].append({"assistant": response})
-
-    # نمایش پیام‌ها
-    for msg in st.session_state["messages"]:
-        if "user" in msg:
-            st.write(f"شما: {msg['user']}")
-        elif "assistant" in msg:
-            st.write(f"دستیار: {msg['assistant']}")
+    # مدیریت درخواست‌های API
+    st.subheader("مدیریت درخواست‌ها")
+    input_data = st.text_input("پیام دریافتی (برای تست):", "")
+    if st.button("تست"):
+        if input_data.strip():
+            st.write("پیام دریافتی:", input_data)
+            response = model(input_data, max_length=50, num_return_sequences=1)
+            st.write("پاسخ مدل:", response[0]["generated_text"])
 
 if __name__ == "__main__":
     main()