Upload app.py

app.py

@@ -27,6 +27,7 @@ from fpdf import FPDF
 import uuid
 import textwrap
 from dotenv import load_dotenv
+import shutil
 try:
     import whisper
 except ImportError:
@@ -39,10 +40,11 @@ whisper_model = whisper.load_model("base")
 load_dotenv()
 
 hf_token = os.getenv("HF_TOKEN")
-resend_api_key = os.getenv("RESEND_API_KEY")
+# resend_api_key = os.getenv("RESEND_API_KEY")
 
 login(token=hf_token)
 
+
 # Mount Google Drive
 #drive.mount('/content/drive')
 
@@ -235,11 +237,13 @@ def process_query(user_query, input_type="text"):
         1. **Diagnosed Mental Disorder**
         2. **Detected emotion** {emotion}
         3. **Intensity of emotion** {value}
-        3. **Matching Symptoms**
-        4. **Personalized Treatment**
+        3. **Matching Symptoms** from the context
+        4. **Personalized Treatment** 
         5. **Helpline Numbers**
         6. **Source Link** (if applicable)
 
+        Make sure to provide a comprehensive and accurate diagnosis and explain the personalised treatment in detail.
+
         If a disorder cannot be determined, return **Diagnosed Mental Disorder** as "Unknown".
 
         ---
@@ -253,11 +257,13 @@ def process_query(user_query, input_type="text"):
     RAG_PROMPT_TEMPLATE = tokenizer.apply_chat_template(
         prompt_in_chat_format, tokenize=False, add_generation_prompt=True
     )
+    
 
     # Generate response
     #answer = READER_LLM(RAG_PROMPT_TEMPLATE)[0]["generated_text"]
     try:
         response = READER_LLM(RAG_PROMPT_TEMPLATE)
+        # print("🔍 Raw LLM output:", response)
         answer = response[0]["generated_text"] if response and "generated_text" in response[0] else "⚠️ No output generated."
     except Exception as e:
         print("❌ Error during generation:", e)
@@ -330,6 +336,8 @@ def show_logs():
 
 def create_summary_pdf(text, filename_prefix="diagnosis_report"):
     try:
+        filename = f"{filename_prefix}_{uuid.uuid4().hex[:6]}.pdf"
+        filepath = os.path.join(".", filename)  # Save in current directory
         pdf = FPDF()
         pdf.add_page()
         pdf.set_font("Arial", style='B', size=14)
@@ -341,12 +349,9 @@ def create_summary_pdf(text, filename_prefix="diagnosis_report"):
         for line in wrapped:
             pdf.cell(200, 10, txt=line, ln=True)
 
-        # Save to /tmp instead of root dir
-        filename = f"/tmp/{filename_prefix}_{uuid.uuid4().hex[:6]}.pdf"
-        pdf.output(filename)
-
-        print(f"✅ PDF created at: {filename}")
-        return filename
+        pdf.output(filepath)
+        print(f"✅ PDF created at: {filepath}")
+        return filepath
     except Exception as e:
         print(f"❌ Error creating PDF: {e}")
         return None
@@ -403,12 +408,16 @@ def unified_handler(audio, text):
         response, download_path = process_query(text, input_type="text")
 
     # Ensure download path is valid
-    if not (download_path and os.path.exists(download_path)):
+    # if not (download_path and os.path.exists(download_path)):
+    #     print("❌ PDF not found or failed to generate.")
+    #     return response, None
+
+    if download_path and os.path.exists(download_path):
+        return response, download_path
+    else:
         print("❌ PDF not found or failed to generate.")
         return response, None
-
-    return response, download_path
-
+        
 #Agentic Framework from HF spaces
 # agent_iframe = gr.HTML(
 #     '<iframe src="https://jaamie-mental-health-agent.hf.space" width="100%" height="700px" style="border:none;"></iframe>'
@@ -482,7 +491,10 @@ feedback_tab = gr.Interface(
         gr.Textbox(label="Confidence Score"),
         gr.Textbox(label="Status")
     ],
-    outputs=gr.Textbox(label="📬 Submission Status"),
+    outputs=[
+    gr.Textbox(label="🧠 Personalized Diagnosis", lines=8),
+    gr.File(label="📥 Download Diagnosis Report")
+    ],
     title="📝 Submit Feedback With Session Metadata"
 )