Update app.py

app.py

@@ -31,36 +31,87 @@ def answer_question(text, question):
     except Exception as e:
         return f"Error in question answering: {str(e)}"
 
-# Gradio interface
-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
+    css=".gradio-container {background: linear-gradient(135deg, #f5f7fa, #c3cfe2);} "
+        ".output-text {font-family: 'Arial', sans-serif; font-size: 16px; color: #333;} "
+        ".input-text {font-family: 'Arial', sans-serif; font-size: 16px; color: #555;} "
+        "button {background: linear-gradient(135deg, #6a11cb, #2575fc); color: white; border: none; padding: 10px 20px; border-radius: 5px;} "
+        "button:hover {background: linear-gradient(135deg, #2575fc, #6a11cb);} "
+) as demo:
+    # Title and description
+    gr.Markdown(
+        """
+        # 🚀 **AI-Powered Study Assistant**
+        **Summarize text, generate flashcards, or answer questions using AI!**
+        """
+    )
 
-# Gradio UI
-interface = gr.Interface(
-    fn=study_assistant,
-    inputs=[
-        gr.Textbox(lines=10, label="Input Text", placeholder="Paste your text here..."),
-        gr.Radio(choices=["Summarize", "Generate Flashcards", "Answer Question"], label="Task"),
-        gr.Textbox(lines=2, label="Question (for Answer Question task)", placeholder="Enter your question here..."),
-    ],
-    outputs=gr.Textbox(label="Output"),
-    title="AI-Powered Study Assistant",
-    description="Summarize text, generate flashcards, or answer questions using AI!",
-    examples=[
-        ["The French Revolution was a period of radical social and political upheaval in France that lasted from 1789 to 1799. It led to the rise of Napoleon Bonaparte and the eventual decline of the French monarchy.", "Summarize"],
-        ["Photosynthesis is the process by which green plants use sunlight to synthesize foods with the help of chlorophyll. It converts carbon dioxide and water into glucose and oxygen.", "Generate Flashcards"],
-        ["The Industrial Revolution began in the 18th century and marked a major turning point in history. Almost every aspect of daily life was influenced in some way.", "Answer Question", "When did the Industrial Revolution begin?"],
-    ],
-)
+    # Inputs
+    with gr.Row():
+        text_input = gr.Textbox(
+            lines=10,
+            label="📝 **Input Text**",
+            placeholder="Paste your text here...",
+            elem_classes="input-text"
+        )
+        question_input = gr.Textbox(
+            lines=2,
+            label="❓ **Question (for Answer Question task)**",
+            placeholder="Enter your question here...",
+            elem_classes="input-text"
+        )
+
+    # Task selection
+    task_radio = gr.Radio(
+        choices=["Summarize", "Generate Flashcards", "Answer Question"],
+        label="🎯 **Task**",
+        value="Summarize"
+    )
+
+    # Output
+    output_text = gr.Textbox(
+        label="📄 **Output**",
+        lines=10,
+        elem_classes="output-text"
+    )
+
+    # Submit button
+    submit_button = gr.Button("✨ **Submit**")
+
+    # Examples
+    gr.Examples(
+        examples=[
+            ["The French Revolution was a period of radical social and political upheaval in France that lasted from 1789 to 1799. It led to the rise of Napoleon Bonaparte and the eventual decline of the French monarchy.", "Summarize"],
+            ["Photosynthesis is the process by which green plants use sunlight to synthesize foods with the help of chlorophyll. It converts carbon dioxide and water into glucose and oxygen.", "Generate Flashcards"],
+            ["The Industrial Revolution began in the 18th century and marked a major turning point in history. Almost every aspect of daily life was influenced in some way.", "Answer Question", "When did the Industrial Revolution begin?"],
+        ],
+        inputs=[text_input, task_radio, question_input],
+        outputs=output_text,
+        fn=study_assistant,
+        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,
+        inputs=[text_input, task_radio, question_input],
+        outputs=output_text
+    )
 
 # Launch the app
-interface.launch()
+demo.launch()