Initial commit

app.py

@@ -107,7 +107,7 @@ def get_user_history(session_id):
         
         history_text = "# 📊 تاریخچه شما\n\n"
         for i, (timestamp, query_type, input_text, response) in enumerate(results, 1):
-            emoji = "💊" if query_type == "drug_interaction" else "🔬"
+            emoji = "💊" if query_type == "drug_interaction" else "🔬" if query_type == "lab_analysis" else "💬"
             history_text += f"## {emoji} مورد {i} - {timestamp}\n\n"
             history_text += f"**ورودی:** {input_text[:100]}...\n\n"
             history_text += f"**پاسخ:** {response[:200]}...\n\n---\n\n"
@@ -130,6 +130,27 @@ def clear_user_history(session_id):
         logger.error(f"Error clearing history: {e}")
         return "❌ خطا در پاک کردن تاریخچه"
 
+def get_chat_history(session_id):
+    """دریافت تاریخچه چت برای حافظه مکالمه"""
+    try:
+        conn = sqlite3.connect("nurca.db")
+        cursor = conn.cursor()
+        cursor.execute('''SELECT input_text, response 
+                         FROM chat_history WHERE session_id = ? AND query_type = 'symptom_chat'
+                         ORDER BY timestamp ASC LIMIT 10''', (session_id,))
+        results = cursor.fetchall()
+        conn.close()
+        
+        chat_messages = []
+        for input_text, response in results:
+            chat_messages.append({"role": "user", "content": input_text})
+            chat_messages.append({"role": "assistant", "content": response})
+        
+        return chat_messages
+    except Exception as e:
+        logger.error(f"Error getting chat history: {e}")
+        return []
+
 def analyze_drug_interaction(drug1, drug2, session_id):
     """تحلیل تداخل دارویی"""
     if not drug1.strip() or not drug2.strip():
@@ -236,10 +257,121 @@ def analyze_lab_image(image, session_id):
         return f"# 🔬 تحلیل نتیجه آزمایش\n\n## متن استخراج شده:\n```\n{extracted_text}\n```\n\n## تحلیل:\n\n{result}"
         
     except Exception as e:
-        error_msg = f" خطا در تحلیل تصویر: {str(e)}"
+        error_msg = f"❌ خطا در تحلیل تصویر: {str(e)}"
         logger.error(error_msg)
         return error_msg
 
+def symptom_diagnosis(symptoms, session_id):
+    """تشخیص بیماری بر اساس علائم"""
+    if not symptoms.strip():
+        return "⚠️ لطفاً علائم خود را توضیح دهید"
+    
+    try:
+        prompt = f"""شما یک پزشک عمومی مهربان و با تجربه هستید. بیمار این علائم را دارد:
+
+علائم: {symptoms}
+
+لطفاً به شکل مهربان و دلگرم کننده موارد زیر را بررسی کنید:
+1. احتمالی‌ترین تشخیص‌ها (با ذکر احتمال)
+2. علائم تکمیلی که باید سؤال شود
+3. راهنمایی‌های خانگی و خودمراقبتی
+4. داروهای بدون نسخه که ممکن است کمک کند
+5. چه زمانی حتماً باید به پزشک مراجعه کند
+6. کلمات آرامش‌بخش و امیدوارانه
+
+⚠️ تأکید کنید که این تشخیص اولیه است و باید با پزشک مشورت شود.
+
+پاسخ را به زبان فارسی، مهربان و کامل بدهید. از کلمات دلگرم کننده استفاده کنید."""
+
+        completion = client.chat.completions.create(
+            extra_headers={
+                "HTTP-Referer": "https://medical-ai.com",
+                "X-Title": "Medical Symptom Analysis",
+            },
+            extra_body={},
+            model="openai/gpt-oss-20b:free",
+            messages=[
+                {
+                    "role": "user",
+                    "content": prompt
+                }
+            ]
+        )
+
+        result = completion.choices[0].message.content
+        
+        # ذخیره در دیتابیس
+        save_to_database(session_id, f"تشخیص علائم: {symptoms}", None, result, "symptom_diagnosis")
+        
+        return f"# 🩺 تشخیص بر اساس علائم\n\n## علائم شما:\n{symptoms}\n\n## نظر پزشک:\n\n{result}"
+        
+    except Exception as e:
+        error_msg = f"❌ خطا در تشخیص: {str(e)}"
+        logger.error(error_msg)
+        return error_msg
+
+def symptom_chat(message, chat_history, session_id):
+    """چت تعاملی برای تشخیص علائم"""
+    if not message.strip():
+        return chat_history, ""
+    
+    try:
+        # دریافت تاریخچه از دیتابیس
+        db_history = get_chat_history(session_id)
+        
+        # ساخت context از تاریخچه
+        context = "تاریخچه مکالمه قبلی:\n"
+        for msg in db_history[-6:]:  # آخرین 6 پیام
+            if msg["role"] == "user":
+                context += f"بیمار: {msg['content']}\n"
+            else:
+                context += f"پزشک: {msg['content']}\n"
+        
+        prompt = f"""شما یک پزشک مهربان هستید که با بیمار چت می‌کنید.
+
+{context}
+
+پیام جدید بیمار: {message}
+
+لطفاً:
+1. به پیام بیمار پاسخ مهربانانه دهید
+2. سؤالات تکمیلی مناسب بپرسید
+3. اگر اطلاعات کافی دارید، تشخیص احتمالی ارائه دهید
+4. همیشه تأکید کنید که باید با پزشک مشورت شود
+
+پاسخ کوتاه و مفید باشد."""
+
+        completion = client.chat.completions.create(
+            extra_headers={
+                "HTTP-Referer": "https://medical-ai.com",
+                "X-Title": "Medical Chat",
+            },
+            extra_body={},
+            model="openai/gpt-oss-20b:free",
+            messages=[
+                {
+                    "role": "user", 
+                    "content": prompt
+                }
+            ]
+        )
+
+        response = completion.choices[0].message.content
+        
+        # ذخیره در دیتابیس
+        save_to_database(session_id, message, None, response, "symptom_chat")
+        
+        # بروزرسانی چت
+        chat_history.append([message, response])
+        
+        return chat_history, ""
+        
+    except Exception as e:
+        error_msg = f"❌ خطا: {str(e)}"
+        logger.error(error_msg)
+        chat_history.append([message, error_msg])
+        return chat_history, ""
+
 # راه‌اندازی دیتابیس
 init_database()
 
@@ -340,6 +472,46 @@ def create_app():
                             label="نتیجه تحلیل"
                         )
             
+            # تب تشخیص علائم
+            with gr.Tab("🩺 تشخیص بر اساس علائم"):
+                with gr.Row():
+                    with gr.Column(scale=2):
+                        gr.HTML('<div class="feature-box"><h3>💬 توضیح علائم شما</h3></div>')
+                        
+                        symptoms_input = gr.Textbox(
+                            label="علائم و نشانه‌های شما",
+                            placeholder="مثال: سردرد شدید، تب، گلودرد، سرفه خشک، ضعف عمومی...",
+                            lines=4
+                        )
+                        
+                        diagnose_btn = gr.Button("🔍 تشخیص علائم", variant="primary", size="lg")
+                    
+                    with gr.Column(scale=3):
+                        symptom_result = gr.Markdown(
+                            value="نتیجه تشخیص اینجا نمایش داده می‌شود...",
+                            label="نتیجه تشخیص"
+                        )
+            
+            # تب چت تشخیص علائم
+            with gr.Tab("💬 چت تشخیص علائم"):
+                with gr.Column():
+                    gr.HTML('<div class="feature-box"><h3>🩺 چت با دستیار پزشکی</h3></div>')
+                    
+                    chatbot = gr.Chatbot(
+                        label="مکالمه با دستیار پزشکی",
+                        height=400
+                    )
+                    
+                    msg_input = gr.Textbox(
+                        label="پیام شما",
+                        placeholder="علائم خود را توضیح دهید یا سؤالات بپرسید...",
+                        lines=2
+                    )
+                    
+                    with gr.Row():
+                        send_btn = gr.Button("📤 ارسال", variant="primary")
+                        clear_chat_btn = gr.Button("🗑️ پاک کردن چت", variant="secondary")
+            
             # تب تاریخچه
             with gr.Tab("📊 تاریخچه و مدیریت"):
                 with gr.Row():
@@ -379,6 +551,29 @@ def create_app():
             outputs=lab_result
         )
         
+        diagnose_btn.click(
+            fn=symptom_diagnosis,
+            inputs=[symptoms_input, session_state],
+            outputs=symptom_result
+        )
+        
+        send_btn.click(
+            fn=symptom_chat,
+            inputs=[msg_input, chatbot, session_state],
+            outputs=[chatbot, msg_input]
+        )
+
+        msg_input.submit(
+            fn=symptom_chat,
+            inputs=[msg_input, chatbot, session_state], 
+            outputs=[chatbot, msg_input]
+        )
+
+        clear_chat_btn.click(
+            fn=lambda: ([], ""),
+            outputs=[chatbot, msg_input]
+        )
+        
         show_history_btn.click(
             fn=get_user_history,
             inputs=session_state,