Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 from langchain.chat_models import ChatOpenAI
 from langchain.schema import HumanMessage
+from docx import Document
+import os
 
 # مدل LLaMA-3 از طریق Together API
 llm = ChatOpenAI(
@@ -49,11 +51,19 @@ def generate_report(operation_data, max_tokens, temperature, top_p):
 
     response = llm(messages)
     
+    # ذخیره گزارش در فایل ورد
+    doc = Document()
+    doc.add_heading('گزارش رسمی عملیات نظامی', 0)
+    doc.add_paragraph(response.content)
     
+    # ذخیره فایل ورد
+    file_path = "military_report.docx"
+    doc.save(file_path)
     
-    return response.content
-    
+    # لینک دانلود فایل ورد
+    return response.content, file_path
 
+# اینترفیس گراڈیو
 demo = gr.Interface(
     fn=generate_report,
     inputs=[
@@ -68,14 +78,15 @@ 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-link")  # نمایش لینک دانلود
     ],
     title="🛰️ سامانه گزارش‌ساز هوش مصنوعی «توانا»",
     description="🔰 اطلاعات خام عملیات را وارد کن تا سامانه توانا یک گزارش رسمی، حرفه‌ای و محرمانه تولید کند.",
     css="""
         body {
-            background-color: #d9dec5 !important;  /* رنگ بک‌گراند جدید */
-            color: #1e293b !important;  /* رنگ متن جدید */
+            background-color: #d9dec5 !important;
+            color: #1e293b !important;
             font-family: 'Vazir', sans-serif;
             direction: rtl;
         }
@@ -84,16 +95,16 @@ demo = gr.Interface(
         }
         label {
             font-size: 22px !important;
-            color: #eeeeee !important;  /* رنگ سفید برای برچسب‌ها */
+            color: #eeeeee !important;
             font-weight: bold;
         }
         #military-input textarea, #military-output textarea {
             background-color: #d9dec5 !important;
             color: #1e293b !important;
-            border: 2px solid #529b39 !important;  /* رنگ جدید برای مرز */
+            border: 2px solid #529b39 !important;
         }
         .gr-button {
-            background-color: #529b39 !important;  /* رنگ دکمه‌ها */
+            background-color: #529b39 !important;
             color: #eeeeee !important;
             border-radius: 12px;
             font-size: 18px;
@@ -101,26 +112,25 @@ demo = gr.Interface(
             transition: all 0.3s ease;
         }
         .gr-button:hover {
-            background-color: #45a12d !important;  /* تغییر رنگ به هنگام hover */
+            background-color: #45a12d !important;
             transform: scale(1.05);
         }
         .gr-button:active {
-            background-color: #3b8e27 !important;  /* رنگ جدید هنگام کلیک */
+            background-color: #3b8e27 !important;
         }
         .gr-button[type="submit"] {
-            background-color: #529b39 !important;  /* رنگ دکمه "تولید گزارش" */
+            background-color: #529b39 !important;
         }
         .gr-button[type="reset"] {
-            background-color: #ff7043 !important;  /* رنگ دکمه "پاک کردن" */
+            background-color: #ff7043 !important;
         }
         .gradio-footer {
             display: none !important;
         }
-        
         .gr-textbox input[type="text"], .gr-textbox textarea {
-            background-color: #1e293b !important;  /* رنگ پس‌زمینه ورودی‌ها */
-            color: #eeeeee !important;  /* رنگ متن ورودی‌ها */
-            border: 2px solid #529b39 !important;  /* رنگ مرز ورودی‌ها */
+            background-color: #1e293b !important;
+            color: #eeeeee !important;
+            border: 2px solid #529b39 !important;
             font-size: 16px;
         }
         .gr-textbox input[type="text"]:focus, .gr-textbox textarea:focus {
@@ -135,4 +145,4 @@ demo = gr.Interface(
 demo.launch()
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()