Update app.py

app.py

@@ -31,6 +31,19 @@ def answer_question(text, question):
     except Exception as e:
         return f"Error in question answering: {str(e)}"
 
+# Function to handle task selection
+def study_assistant(text, task, question=None):
+    if task == "Summarize":
+        return summarize(text)
+    elif task == "Generate Flashcards":
+        return generate_flashcards(text)
+    elif task == "Answer Question":
+        if not question:
+            return "Please enter a question."
+        return answer_question(text, question)
+    else:
+        return "Invalid task selected."
+
 # Gradio Blocks for advanced UI
 with gr.Blocks(
     theme=gr.themes.Soft(primary_hue="teal", secondary_hue="pink"),  # Use a vibrant theme
@@ -93,19 +106,6 @@ with gr.Blocks(
         label="📚 **Examples**"
     )
 
-    # Function to handle task selection
-    def study_assistant(text, task, question=None):
-        if task == "Summarize":
-            return summarize(text)
-        elif task == "Generate Flashcards":
-            return generate_flashcards(text)
-        elif task == "Answer Question":
-            if not question:
-                return "Please enter a question."
-            return answer_question(text, question)
-        else:
-            return "Invalid task selected."
-
     # Link button to function
     submit_button.click(
         study_assistant,