Update app.py

app.py

@@ -2,6 +2,7 @@ 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",
@@ -9,6 +10,7 @@ llm = ChatOpenAI(
     max_tokens=1024
 )
 
+# تابع تولید گزارش
 def generate_report(operation_data, max_tokens, temperature, top_p):
     system_prompt = """تو یک افسر گزارش‌نویس نظامی هستی. وظیفه تو تهیه یک گزارش رسمی، دقیق، ساختاریافته و حرفه‌ای از عملیات نظامی بر اساس داده‌های ارائه‌شده است. اگر اطلاعاتی در متن وجود نداشت، صراحتاً بنویس «ذکر نشده است».
 لطفاً گزارش را دقیقاً در قالب فرمتیافته زیر تولید کن، با استفاده از نشانه‌گذاری واضح (مانند ایموجی‌ها و خطوط جداکننده) تا خروجی قابل‌خواندن و رسمی باشد:
@@ -40,36 +42,33 @@ def generate_report(operation_data, max_tokens, temperature, top_p):
 ✅ این گزارش بر اساس اطلاعات خام ارائه‌شده تولید شده و ممکن است برخی جزئیات در دسترس نبوده باشند.
 """
 
+
+    # تعریف پیام‌ها برای مدل
     messages = [
         HumanMessage(content=f"{system_prompt}\n\n{operation_data}")
     ]
 
+    # ارسال پیام به مدل و گرفتن پاسخ
     response = llm(messages)
     return response.content
 
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("### 🛰️ سامانه گزارش‌ساز هوش مصنوعی «توانا»")
-    gr.Markdown("🔰 اطلاعات خام عملیات را وارد کن تا سامانه توانا یک گزارش رسمی، حرفه‌ای و محرمانه تولید کند.")
-    
-    with gr.Row():
-        operation_data = gr.Textbox(
+demo = gr.Interface(
+    fn=generate_report,
+    inputs=[
+        gr.Textbox(
             label="🪖 اطلاعات عملیات نظامی",
             lines=10,
             placeholder="مثلاً: در ساعت ۵ صبح، گردان الف از محور غربی وارد منطقه شد...",
             elem_id="military-input"
-        )
-        
-        max_tokens = gr.Slider(128, 2048, value=1024, step=64, label="🎯 حداکثر توکن خروجی", elem_id="slider-1")
-        temperature = gr.Slider(0.1, 1.5, value=0.7, step=0.1, label="🔥 دمای خلاقیت (temperature)", elem_id="slider-2")
-        top_p = gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="🎯 Top-p", elem_id="slider-3")
-    
-    with gr.Row():
-        generate_btn = gr.Button("🎯 تولید گزارش", elem_id="generate-btn")
-        clear_btn = gr.Button("🗑️ پاک کردن", elem_id="clear-btn")
-
-    military_output = gr.Textbox(label="📄 گزارش رسمی تولید شده", lines=10, elem_id="military-output")
-
-    demo.css = """
+        ),
+        gr.Slider(128, 2048, value=1024, step=64, label="🎯 حداکثر توکن خروجی", elem_id="slider-1"),
+        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"),
+    title="🛰️ سامانه گزارش‌ساز هوش مصنوعی «توانا»",
+    description="🔰 اطلاعات خام عملیات را وارد کن تا سامانه توانا یک گزارش رسمی، حرفه‌ای و محرمانه تولید کند.",
+    css="""
         body {
             background-color: #d9dec5 !important;  /* رنگ بک‌گراند جدید */
             color: #1e293b !important;  /* رنگ متن جدید */
@@ -124,8 +123,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             border-color: #FF6F00 !important;
             outline: none;
         }
-    """
-
+    """,
+    submit_btn="🎯 تولید گزارش",
+    clear_btn="🗑️ پاک کردن"
+)
 demo.launch()
 
 if __name__ == "__main__":