app.py

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",
    model="meta-llama/Llama-3.3-70B-Instruct-Turbo-Free",
    max_tokens=1024
)

# تابع تولید گزارش
def generate_report(operation_data, max_tokens, temperature, top_p):
    system_prompt = """تو یک افسر گزارش‌نویس نظامی هستی. وظیفه تو تهیه یک گزارش رسمی، دقیق، ساختاریافته و حرفه‌ای از عملیات نظامی بر اساس داده‌های ارائه‌شده است. اگر اطلاعاتی در متن وجود نداشت، صراحتاً بنویس «ذکر نشده است».
لطفاً گزارش را دقیقاً در قالب فرمتیافته زیر تولید کن، با استفاده از نشانه‌گذاری واضح (مانند ایموجی‌ها و خطوط جداکننده) تا خروجی قابل‌خواندن و رسمی باشد:
━━━━━━━━━━━━━━━━━━━━━━
📄 گزارش رسمی عملیات نظامی
🕒 تاریخ و ساعت شروع عملیات:
[اگر در متن بود بنویس، در غیر این صورت بنویس: ذکر نشده است]
🕓 تاریخ و ساعت پایان عملیات:
[اگر در متن بود بنویس، در غیر این صورت بنویس: ذکر نشده است]
📍 مکان عملیات:
[نام مکان عملیات یا بنویس: ذکر نشده است]
🪖 واحدهای شرکت‌کننده:
[نام گردان‌ها یا یگان‌ها یا بنویس: ذکر نشده است]
🎯 هدف عملیات:
[اهداف اصلی عملیات یا بنویس: ذکر نشده است]
📝 شرح عملیات:
[شرح مختصر عملیات شامل مسیر، تاکتیک، درگیری‌ها و روند پیشروی]
🏁 نتیجه عملیات:
[پیروزی، شکست، توقف، عقب‌نشینی یا ذکر نشده است]
🩻 تلفات نیروهای خودی:
[تعداد شهدا، مجروحین یا بنویس: ذکر نشده است]
🚫 تلفات دشمن:
[در صورت موجود بودن، ذکر شود - در غیر این صورت بنویس: ذکر نشده است]
🛠️ تجهیزات استفاده‌شده یا از دست‌رفته:
[مثلاً تانک، پهپاد، سلاح خاص یا ذکر نشده است]
🗒️ یادداشت‌های اضافی:
[شرایط خاص، اطلاعات پشتیبانی، آب‌وهوا، یا بنویس: ذکر نشده است]
━━━━━━━━━━━━━━━━━━━━━━
✅ این گزارش بر اساس اطلاعات خام ارائه‌شده تولید شده و ممکن است برخی جزئیات در دسترس نبوده باشند.
"""

    # تعریف پیام‌ها برای مدل
    messages = [
        HumanMessage(content=f"{system_prompt}\n\n{operation_data}")
    ]

    # ارسال پیام به مدل و گرفتن پاسخ
    response = llm(messages)
    
    # ذخیره گزارش در یک فایل متنی
    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,
    inputs=[
        gr.Textbox(
            label="🪖 اطلاعات عملیات نظامی",
            lines=10,
            placeholder="مثلاً: در ساعت ۵ صبح، گردان الف از محور غربی وارد منطقه شد...",
            elem_id="military-input"
        ),
        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"),
        gr.File(label=" دانلود گزارش", elem_id="download-btn", visible=False)  # لینک دانلود مخفی است تا وقتی گزارش نوشته شده
    ],
    title="🛰️ سامانه گزارش‌ساز هوش مصنوعی «توانا»",
    description="🔰 اطلاعات خام عملیات را وارد کن تا سامانه توانا یک گزارش رسمی، حرفه‌ای و محرمانه تولید کند.",
    css="""
        body {
            background-color: #d9dec5 !important;  /* رنگ بک‌گراند جدید */
            color: #1e293b !important;  /* رنگ متن جدید */
            font-family: 'Vazir', sans-serif;
            direction: rtl;
        }
        .gradio-container {
            background-color: #d9dec5 !important;
        }
        label {
            font-size: 22px !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;  /* رنگ جدید برای مرز */
        }
        .gr-button {
            background-color: #529b39 !important;  /* رنگ دکمه‌ها */
            color: #eeeeee !important;
            border-radius: 12px;
            font-size: 18px;
            padding: 14px;
            transition: all 0.3s ease;
        }
        .gr-button:hover {
            background-color: #45a12d !important;  /* تغییر رنگ به هنگام hover */
            transform: scale(1.05);
        }
        .gr-button:active {
            background-color: #3b8e27 !important;  /* رنگ جدید هنگام کلیک */
        }
        .gr-button[type="submit"] {
            background-color: #529b39 !important;  /* رنگ دکمه "تولید گزارش" */
        }
        .gr-button[type="reset"] {
            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;  /* رنگ مرز ورودی‌ها */
            font-size: 16px;
        }
        .gr-textbox input[type="text"]:focus, .gr-textbox textarea:focus {
            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__":
    demo.launch()