Update app.py

app.py

@@ -42,7 +42,6 @@ def generate_report(operation_data, max_tokens, temperature, top_p):
 ✅ این گزارش بر اساس اطلاعات خام ارائه‌شده تولید شده و ممکن است برخی جزئیات در دسترس نبوده باشند.
 """
 
-
     # تعریف پیام‌ها برای مدل
     messages = [
         HumanMessage(content=f"{system_prompt}\n\n{operation_data}")
@@ -50,7 +49,12 @@ def generate_report(operation_data, max_tokens, temperature, top_p):
 
     # ارسال پیام به مدل و گرفتن پاسخ
     response = llm(messages)
-    return response.content
+    
+    # ذخیره گزارش در یک فایل متنی
+    with open("report.txt", "w", encoding="utf-8") as file:
+        file.write(response.content)
+    
+    return response.content, "report.txt"  # لینک به فایل متنی برای دانلود
 
 demo = gr.Interface(
     fn=generate_report,
@@ -65,7 +69,10 @@ demo = gr.Interface(
         gr.Slider(0.1, 1.5, value=0.7, step=0.1, label="🔥 دمای خلاقیت (temperature)", elem_id="slider-2"),
         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"),
+    outputs=[
+        gr.Textbox(label="📄 گزارش رسمی تولید شده", lines=10, elem_id="military-output"),
+        gr.File(label=" دانلود گزارش", elem_id="download-btn", visible=False)  # لینک دانلود مخفی است تا وقتی گزارش نوشته شده
+    ],
     title="🛰️ سامانه گزارش‌ساز هوش مصنوعی «توانا»",
     description="🔰 اطلاعات خام عملیات را وارد کن تا سامانه توانا یک گزارش رسمی، حرفه‌ای و محرمانه تولید کند.",
     css="""
@@ -123,10 +130,52 @@ demo = gr.Interface(
             border-color: #FF6F00 !important;
             outline: none;
         }
+
+        /* استایل جدید برای دکمه دانلود به شکل کارت */
+        #download-btn {
+            background-color: #FF6F00 !important;  /* رنگ پس‌زمینه کارت نارنجی */
+            color: #eeeeee !important;  /* رنگ متن */
+            border-radius: 12px;
+            padding: 10px;
+            box-shadow: 0 4px 6px rgba(0, 0, 0, 0.2);
+            font-size: 16px;
+            font-weight: bold;
+            text-align: center;
+            transition: all 0.3s ease;
+            display: inline-block;
+            margin-top: 20px;
+            cursor: pointer;
+            width: 100%;
+            max-width: 200px;
+            text-decoration: none;
+            border: none;
+        }
+
+        #download-btn:hover {
+            background-color: #e65c00 !important;  /* تغییر رنگ به هنگام hover */
+            transform: scale(1.05);
+            box-shadow: 0 6px 8px rgba(0, 0, 0, 0.3);
+        }
+
+        #download-btn:active {
+            background-color: #cc5200 !important;  /* رنگ پس‌زمینه هنگام کلیک */
+            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
+        }
+
+        /* نمایش کارت دانلود فقط بعد از تولید گزارش */
+        .gr-file {
+            visibility: hidden;
+        }
     """,
     submit_btn="🎯 تولید گزارش",
     clear_btn="🗑️ پاک کردن"
 )
+
+def update_download_button(response, file):
+    if response:
+        return gr.File.update(visible=True, value=file)
+    return gr.File.update(visible=False)
+
 demo.launch()
 
 if __name__ == "__main__":