Update app.py

app.py

@@ -2,21 +2,34 @@ import gradio as gr
 import random
 from PIL import Image
 
-from funcs import detect_emotions, cosine_distance, generate_triggers_img, session_processor
+from funcs import detect_emotions, cosine_distance, generate_triggers_img, get_doc_response_emotions, summarize_and_recommend
 
-sp = session_processor
+# img_paths = ['2.jpg','3.jpeg','4.jpeg','5.jpeg','6.jpeg']
+# img_path = random.choice(img_paths)
+# img_1 = Image.open(img_path)
 
-img_paths = ['2.jpg','3.jpeg','4.jpeg','5.jpeg','6.jpeg']
-img_path = random.choice(img_paths)
-img_1 = Image.open(img_path)
+# img_path = '7.jpg'
+# img_2 = Image.open(img_path)
 
-img_path = '7.jpg'
-img_2 = Image.open(img_path)
+def submit(user_message, therapy_session_conversation, session_conversation):
+    response = get_doc_response_emotions(user_message, therapy_session_conversation)
+    return '', therapy_session_conversation, emotions_msg
 
+def sum_rec(session_conversation):
+    full_summary, full_recommendations = summarize_and_recommend(session_conversation)
+    return full_summary, full_recommendations
+    
 with gr.Blocks() as demo:
+    img_paths = ['2.jpg','3.jpeg','4.jpeg','5.jpeg','6.jpeg']
+    img_path = random.choice(img_paths)
+    img_1 = Image.open(img_path)
+    
+    img_path = '7.jpg'
+    img_2 = Image.open(img_path)
+    
     gr.Markdown("""# Falcon Cognitive Behavioural Therapy Assistant
     - Your Personal AI Therapist. Start chatting...""")
-    therapy_session_conversation = gr.State([])
+    session_conversation = gr.State([])
 
     with gr.Row():
 
@@ -29,17 +42,17 @@ with gr.Blocks() as demo:
             chatbox = gr.Chatbot(label="Therapy Session Conversation",value =[[None, 'Therapist: Hello, What can I do for you?']], height=300)
             user_input = gr.Textbox(placeholder="Enter your message here...", label="User")
             submit_button = gr.Button("Submit")
-            submit_button.click(sp.get_doc_response_emotions, [user_input, chatbox], [user_input, chatbox, emotions])
-            user_input.submit(sp.get_doc_response_emotions, [user_input, chatbox], [user_input, chatbox, emotions])
+            submit_button.click(submit, [user_input, chatbox], [user_input, chatbox, emotions])
+            user_input.submit(submit, [user_input, chatbox], [user_input, chatbox, emotions])
             recommendations = gr.Textbox(label="Recommended Actions", visible = False)
 
             def summarize_and_recommend_process():
-              sn, r = sp.summarize_and_recommend()
+              sn, r = summarize_and_recommend()
               return gr.update(visible=True, value=sn), gr.update(visible=True, value=r)
               
     process_button = gr.Button("Generate Session Notes & Recommendations")
     clear = gr.ClearButton(components=[user_input, chatbox, emotions, summary_notes, recommendations], value="Clear console")
-    process_button.click(summarize_and_recommend_process, inputs=None, outputs=[summary_notes, recommendations])
+    process_button.click(sum_rec, inputs=None, outputs=[summary_notes, recommendations])
 
 
 demo.launch(debug=True, share=True)