Update app.py

app.py

@@ -1,8 +1,8 @@
-
 import gradio as gr
 from langchain.chat_models import ChatOpenAI
 from langchain.schema import HumanMessage
 
+# مدل LLaMA-3 از طریق Together API
 llm = ChatOpenAI(
     base_url="https://api.together.xyz/v1",
     api_key="0291f33aee03412a47fa5d8e562e515182dcc5d9aac5a7fb5eefdd1759005979",
@@ -10,6 +10,7 @@ llm = ChatOpenAI(
     max_tokens=1024
 )
 
+# تابع تولید گزارش
 def generate_report(operation_data, max_tokens, temperature, top_p):
     system_prompt = """تو یک افسر گزارش‌نویس نظامی هستی. وظیفه تو تهیه یک گزارش رسمی، دقیق، ساختاریافته و حرفه‌ای از عملیات نظامی بر اساس داده‌های ارائه‌شده است. اگر اطلاعاتی در متن وجود نداشت، صراحتاً بنویس «ذکر نشده است».
 لطفاً گزارش را دقیقاً در قالب فرمتیافته زیر تولید کن، با استفاده از نشانه‌گذاری واضح (مانند ایموجی‌ها و خطوط جداکننده) تا خروجی قابل‌خواندن و رسمی باشد:
@@ -41,19 +42,19 @@ def generate_report(operation_data, max_tokens, temperature, top_p):
 ✅ این گزارش بر اساس اطلاعات خام ارائه‌شده تولید شده و ممکن است برخی جزئیات در دسترس نبوده باشند.
 """
 
-
+    # تعریف پیام‌ها برای مدل
     messages = [
         HumanMessage(content=f"{system_prompt}\n\n{operation_data}")
     ]
 
-    response = llm.invoke(messages)
-    report_text = response.content
+    # ارسال پیام به مدل و گرفتن پاسخ
+    response = llm(messages)
     
+    # ذخیره گزارش در یک فایل متنی
     with open("report.txt", "w", encoding="utf-8") as file:
         file.write(response.content)
     
-    return response.content
-
+    return response.content, "report.txt"  # لینک به فایل متنی برای دانلود
 
 demo = gr.Interface(
     fn=generate_report,
@@ -69,7 +70,8 @@ demo = gr.Interface(
         gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="🎯 Top-p", elem_id="slider-3"),
     ],
     outputs=[
-        gr.Textbox(label="📄 گزارش رسمی تولید شده", lines=10, elem_id="military-output")
+        gr.Textbox(label="📄 گزارش رسمی تولید شده", lines=10, elem_id="military-output"),
+        gr.File(label=" دانلود گزارش", elem_id="download-btn")  # اضافه کردن دکمه دانلود
     ],
     title="🛰️ سامانه گزارش‌ساز هوش مصنوعی «توانا»",
     description="🔰 اطلاعات خام عملیات را وارد کن تا سامانه توانا یک گزارش رسمی، حرفه‌ای و محرمانه تولید کند.",
@@ -133,7 +135,7 @@ demo = gr.Interface(
     clear_btn="🗑️ پاک کردن"
 )
 
-demo.launch(share=True)
+demo.launch()
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch()