Update app.py

app.py

@@ -257,32 +257,23 @@ if __name__ == "__main__":
     flask_thread.start()
 
     # Gradio interface
-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": ""})
+    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": ""}) 
 
         # 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])