Update app.py

app.py

@@ -29,29 +29,49 @@ def study_aid(question, context, font_size=16, audio_output=False, simplify_text
         audio = tts(answer)
         return output, audio
     
-    return output
+    return output, None
 
 def submit_feedback(feedback):
     with open("feedback.txt", "a") as f:
         f.write(feedback + "\n")
     return "Feedback submitted!"
 
-# Create Gradio interface
-iface = gr.Interface(
-    fn=[study_aid, submit_feedback],
-    inputs=[
-        [
-            gr.Textbox(label="Question", placeholder="e.g., What is machine learning?"),
-            gr.Textbox(label="Context (Lecture Notes)", placeholder="Paste your notes here..."),
-            gr.Slider(12, 24, value=16, label="Font Size (px)"),
-            gr.Checkbox(label="Generate Audio Output"),
-            gr.Checkbox(label="Simplify Text")
-        ],
-        gr.Textbox(label="Feedback", placeholder="Report issues or suggestions...")
-    ],
-    outputs=["html", "audio", "text"],
-    title="StudyBuddy: Accessible Study Aid for Neurodiverse Students",
-    description="Ask questions about your college lecture notes with accessible text and audio outputs. No data is stored."
-)
+# Create Gradio app with Blocks
+with gr.Blocks(title="StudyBuddy: Accessible Study Aid for Neurodiverse Students") as app:
+    gr.Markdown(
+        """
+        # StudyBuddy: Accessible Study Aid for Neurodiverse Students
+        Ask questions about your college lecture notes with accessible text and audio outputs. No data is stored.
+        """
+    )
+    
+    # Tab 1: Study Aid
+    with gr.Tab("Ask a Question"):
+        question_input = gr.Textbox(label="Question", placeholder="e.g., What is machine learning?")
+        context_input = gr.Textbox(label="Context (Lecture Notes)", placeholder="Paste your notes here...")
+        font_size_input = gr.Slider(12, 24, value=16, label="Font Size (px)")
+        audio_output_input = gr.Checkbox(label="Generate Audio Output")
+        simplify_text_input = gr.Checkbox(label="Simplify Text")
+        study_submit_btn = gr.Button("Get Answer")
+        study_output_text = gr.HTML(label="Answer")
+        study_output_audio = gr.Audio(label="Audio Narration")
+        
+        study_submit_btn.click(
+            fn=study_aid,
+            inputs=[question_input, context_input, font_size_input, audio_output_input, simplify_text_input],
+            outputs=[study_output_text, study_output_audio]
+        )
+    
+    # Tab 2: Feedback
+    with gr.Tab("Submit Feedback"):
+        feedback_input = gr.Textbox(label="Feedback", placeholder="Report issues or suggestions...")
+        feedback_submit_btn = gr.Button("Submit Feedback")
+        feedback_output = gr.Text(label="Feedback Status")
+        
+        feedback_submit_btn.click(
+            fn=submit_feedback,
+            inputs=feedback_input,
+            outputs=feedback_output
+        )
 
-iface.launch()
+app.launch()