Update app.py

app.py

@@ -257,23 +257,32 @@ if __name__ == "__main__":
     flask_thread.start()
 
     # Gradio interface
-    with gr.Blocks() as interface:
-        gr.Markdown("""<div style="text-align:center; margin-top:20px;"><h1>PainSense: AI-Driven Pain Detection and Chatbot Assistance</h1></div>""")
-        with gr.Row():
-            with gr.Column(scale=1):
-                file_input = gr.File(label="Upload Audio or Video File")
-                with gr.Row():  # Place buttons next to each other
-                    clear_button = gr.Button("Clear", elem_id="clear_btn")
-                    submit_button = gr.Button("Submit", variant="primary", elem_id="submit_button")
-                chatbot_input = gr.Textbox(label="Chat with AI", placeholder="Ask a question...", interactive=True)
-                chatbot_output = gr.Textbox(label="Chatbot Response", interactive=False)
-
-            with gr.Column(scale=1):
-                pain_result = gr.Textbox(label="Pain Detection Result")
-                average_pain = gr.Textbox(label="Average Pain")
-                graph_output = gr.Image(label="Pain Detection Graph")
-
-        state = gr.State({"result": "", "average_pain": "", "graph_filename": ""})
+with gr.Blocks() as interface:
+    # Title
+    gr.Markdown("""
+    <div style="text-align:center; margin-top:20px;">
+        <h1>PainSense: AI-Driven Pain Detection and Chatbot Assistance</h1>
+    </div>
+    """)
+
+    with gr.Row():
+        # Left Column
+        with gr.Column(scale=1):
+            file_input = gr.File(label="Upload Audio or Video File", file_types=["audio", "video"])
+            with gr.Row():  # Place buttons side by side
+                clear_button = gr.Button("Clear", elem_id="clear_btn")
+                submit_button = gr.Button("Submit", variant="primary", elem_id="submit_button")
+            chatbot_input = gr.Textbox(label="Chat with AI", placeholder="Ask a question...", interactive=True)
+            chatbot_output = gr.Textbox(label="Chatbot Response", interactive=False)
+
+        # Right Column
+        with gr.Column(scale=1):
+            pain_result = gr.Textbox(label="Pain Detection Result", interactive=False)
+            average_pain = gr.Textbox(label="Average Pain", interactive=False)
+            graph_output = gr.Image(label="Pain Detection Graph")
+
+    # State to store intermediate values
+    state = gr.State({"result": "", "average_pain": "", "graph_filename": ""})
 
         # Clear button resets the UI, including the file input, chatbot input, and outputs
         clear_button.click(lambda: (None, None, "", "",""), outputs=[pain_result, average_pain, graph_output, chatbot_output, file_input])