Upload app.py

Linked agentic framework with the Langchain framework

app.py

@@ -40,8 +40,6 @@ whisper_model = whisper.load_model("base")
 load_dotenv()
 
 hf_token = os.getenv("HF_TOKEN")
-# resend_api_key = os.getenv("RESEND_API_KEY")
-
 login(token=hf_token)
 
 
@@ -71,7 +69,7 @@ feedback_file_path = "./feedback_logs.csv"
 if not os.path.exists(log_file_path):
     with open(log_file_path, "w", newline="", encoding="utf-8") as f:
       writer = csv.writer(f)
-      writer.writerow(["timestamp", "input_type", "query", "diagnosis", "confidence_score", "status"])
+      writer.writerow(["timestamp", "user_id", "input_type", "query", "diagnosis", "confidence_score", "status"])
 
 # -------------------------------
 # 🔧 Feedback setup
@@ -80,10 +78,11 @@ if not os.path.exists(feedback_file_path):
     with open(feedback_file_path, "w", newline="", encoding="utf-8") as f:
         writer = csv.writer(f)
         writer.writerow([
-            "feedback_id", "timestamp", "input_type", "query",
-            "diagnosis", "confidence_score", "status", "feedback"
+            "feedback_id", "timestamp", "user_id", "input_type", "query",
+            "diagnosis", "status", "feedback"
         ])
 
+
 # Ensure directory exists
 #os.makedirs(save_path_bge, exist_ok=True)
 
@@ -280,9 +279,17 @@ def process_query(user_query, input_type="text"):
 
     # Log interaction
     log_query(input_type=input_type, query=user_query, diagnosis=diagnosis, confidence_score=severity_score, status=status)
-    download_path = create_summary_pdf(answer)
+    download_path = create_summary_txt(answer)
+
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    user_id = session_data["latest"]["user_id"]  # grab it from session
 
-    return answer, download_path
+    # Prepend to the answer string
+    answer_header = f"🧾 Session ID: {user_id}\n📅 Timestamp: {timestamp}\n\n"
+    return answer_header + answer, download_path
+
+
+    #return answer, download_path
 
 # Dashboard Interface
 def diagnosis_dashboard():
@@ -315,17 +322,47 @@ def diagnosis_dashboard():
         return f"⚠️ Error loading dashboard: {str(e)}"
 
 # For logs functionality
+# def log_query(input_type, query, diagnosis, confidence_score, status):
+#     with open(log_file_path, "a", newline="", encoding="utf-8") as f:
+#         writer = csv.writer(f, quoting=csv.QUOTE_ALL)
+#         writer.writerow([
+#             datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+#             input_type.replace('"', '""'),
+#             query.replace('"', '""'),
+#             diagnosis.replace('"', '""'),
+#             str(confidence_score),
+#             status
+#         ])
+
+session_data = {}
+
 def log_query(input_type, query, diagnosis, confidence_score, status):
+    
+    user_id = f"SSuser_ID_{uuid.uuid4().hex[:8]}"
+
+    # Store in-memory session data for feedback use
+    session_data["latest"] = {
+        "user_id": user_id,
+        "input_type": input_type,
+        "query": query,
+        "diagnosis": diagnosis,
+        "confidence_score": confidence_score,
+        "status": status
+    }
+
     with open(log_file_path, "a", newline="", encoding="utf-8") as f:
         writer = csv.writer(f, quoting=csv.QUOTE_ALL)
         writer.writerow([
-            datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
-            input_type.replace('"', '""'),
-            query.replace('"', '""'),
-            diagnosis.replace('"', '""'),
+            str(datetime.now().strftime("%Y-%m-%d %H:%M:%S")),
+            str(user_id),
+            str(input_type).replace('"', '""'),
+            str(query).replace('"', '""'),
+            str(diagnosis).replace('"', '""'),
             str(confidence_score),
-            status
+            str(status)
         ])
+
+
 def show_logs():
     try:
         df = pd.read_csv(log_file_path)
@@ -357,18 +394,45 @@ def create_summary_pdf(text, filename_prefix="diagnosis_report"):
         return None
 
 
-def create_text_file(content, filename_prefix="diagnosis_text"):
+
+
+def create_summary_txt(text, filename_prefix="diagnosis_report"):
     filename = f"{filename_prefix}_{uuid.uuid4().hex[:6]}.txt"
     with open(filename, "w", encoding="utf-8") as f:
-        f.write(content)
+        f.write(text)
+    print(f"✅ TXT report created: {filename}")
     return filename
 
 
 
 # 📥 Feedback
-feedback_data = []
-def submit_feedback(feedback, input_type, query, diagnosis, confidence_score, status):
-    feedback_id = str(uuid.uuid4())
+# feedback_data = []
+# def submit_feedback(feedback, input_type, query, diagnosis, confidence_score, status):
+#     feedback_id = str(uuid.uuid4())
+#     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+
+#     with open(feedback_file_path, "a", newline="", encoding="utf-8") as f:
+#         writer = csv.writer(f, quoting=csv.QUOTE_ALL)
+#         writer.writerow([
+#             feedback_id,
+#             timestamp,
+#             input_type.replace('"', '""'),
+#             query.replace('"', '""'),
+#             diagnosis.replace('"', '""'),
+#             str(confidence_score),
+#             status,
+#             feedback.replace('"', '""')
+#         ])
+
+#     return f"✅ Feedback received! Your Feedback ID: {feedback_id}"
+
+
+def submit_feedback(feedback):
+    # if "latest" not in session_data:
+    #     return "⚠️ No diagnosis found for this session. Please get a diagnosis first."
+
+    user_info = session_data["latest"]
+    feedback_id = f"fb_{uuid.uuid4().hex[:8]}"
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
 
     with open(feedback_file_path, "a", newline="", encoding="utf-8") as f:
@@ -376,11 +440,11 @@ def submit_feedback(feedback, input_type, query, diagnosis, confidence_score, st
         writer.writerow([
             feedback_id,
             timestamp,
-            input_type.replace('"', '""'),
-            query.replace('"', '""'),
-            diagnosis.replace('"', '""'),
-            str(confidence_score),
-            status,
+            user_info["user_id"],
+            user_info["input_type"],
+            user_info["query"],
+            user_info["diagnosis"],
+            user_info["status"],
             feedback.replace('"', '""')
         ])
 
@@ -400,24 +464,37 @@ def download_feedback_log():
 #     })
 #     return "✅ Diagnosis report sent to your email!" if response.get("id") else "⚠️ Failed to send email."
 
+# For pdf
+
+# def unified_handler(audio, text):
+#     if audio:
+#         response, download_path = process_whisper_query(audio)
+#     else:
+#         response, download_path = process_query(text, input_type="text")
+
+#     # Ensure download path is valid
+#     # 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
+
+# for text doc download
 
 def unified_handler(audio, text):
     if audio:
-        response, download_path = process_whisper_query(audio)
+        response, _ = process_whisper_query(audio)
     else:
-        response, download_path = process_query(text, input_type="text")
+        response, _ = process_query(text, input_type="text")
 
-    # Ensure download path is valid
-    # if not (download_path and os.path.exists(download_path)):
-    #     print("❌ PDF not found or failed to generate.")
-    #     return response, None
+    download_path = create_summary_txt(response)  # <- save as txt instead
+
+    return response, download_path
 
-    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
-        
 #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>'
@@ -440,7 +517,7 @@ main_assistant_tab = gr.Interface(
         gr.Textbox(lines=2, placeholder="Or type your mental health concern here...")
     ],
     outputs=[
-        gr.Textbox(label="🧠 Personalized Diagnosis", lines=8),
+        gr.Textbox(label="🧠 Personalized Diagnosis", lines=15, show_copy_button=True),
         gr.File(label="📥 Download Diagnosis Report")
     ],
     title="🧠 SafeSpace AI",
@@ -478,28 +555,19 @@ logs_tab = gr.Interface(
 #     title="📝 Submit Feedback With Session Metadata"
 # )
 
-def feedback_handler(fb, inp_type, query, diag, score, status):
-    return submit_feedback(fb, inp_type, query, diag, score, status)
+# def feedback_handler(fb, inp_type, query, diag, score, status):
+#     return submit_feedback(fb, inp_type, query, diag, score, status)
 
 feedback_tab = gr.Interface(
-    fn=feedback_handler,
-    inputs=[
-        gr.Textbox(label="📝 Feedback"),
-        gr.Textbox(label="Input Type"),
-        gr.Textbox(label="Query"),
-        gr.Textbox(label="Diagnosis"),
-        gr.Textbox(label="Confidence Score"),
-        gr.Textbox(label="Status")
-    ],
-    outputs=[
-    gr.Textbox(label="🧠 Personalized Diagnosis", lines=8),
-    gr.File(label="📥 Download Diagnosis Report")
-    ],
-    title="📝 Submit Feedback With Session Metadata"
+    fn=submit_feedback,
+    inputs=[gr.Textbox(label="📝 Share your thoughts")],
+    outputs="text",
+    title="📝 Submit Feedback"
 )
 
 
 
+
 feedback_download_tab = gr.Interface(
     fn=download_feedback_log,
     inputs=[],
@@ -508,14 +576,19 @@ feedback_download_tab = gr.Interface(
 )
 
 agent_tab = gr.Interface(
-    fn=lambda: "Use the assistant below 👇",
+    fn=lambda: "",
     inputs=[],
-    outputs=gr.Markdown(
-        f'<iframe src="https://jaamie-mental-health-agent.hf.space" width="100%" height="700px" style="border:none;"></iframe>'
+    outputs=gr.HTML(
+        """<button onclick="window.open('https://jaamie-mental-health-agent.hf.space', '_blank')" 
+           style='padding:10px 20px; font-size:16px; background-color:#4CAF50; color:white; border:none; border-radius:5px;'>
+           🧠 Launch Agent SafeSpace 001
+        </button>"""
     ),
-    title="🤖 Agent Portal"
+    title="🤖 Agent SafeSpace 001"
 )
 
+
+
 # Add to your tab list
 app = gr.TabbedInterface(
     interface_list=[
@@ -524,7 +597,7 @@ app = gr.TabbedInterface(
         logs_tab,
         feedback_tab,
         feedback_download_tab,
-        agent_tab  # ⬅️ Add this!
+        agent_tab 
     ],
     tab_names=[
         "🧠 Assistant",
@@ -532,7 +605,7 @@ app = gr.TabbedInterface(
         "📄 Logs",
         "📝 Feedback",
         "📂 Feedback CSV",
-        "🤖 Agent"
+        "🤖 Agent 001"
     ]
 )