Update app.py

app.py

@@ -82,15 +82,21 @@ def generate_output(prompt_type, query):
     chain = LLMChain(llm=llm, prompt=templates[prompt_type])
     return chain.run({"context": context, "query": query})
 
-# Gradio UI
+# ✅ Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("# 📐 Geometry Assistant")
+    gr.Markdown("# 📐 Geometry Teaching Assistant")
 
-    query = gr.Textbox(label="Enter a geometry topic")
-    prompt_type = gr.Dropdown(["flashcard"], value="flashcard", label="Prompt Type")
-    output = gr.Textbox(label="Generated Output", lines=12)
+    with gr.Row():
+        query = gr.Textbox(label="Enter a geometry topic")
+        prompt_type = gr.Dropdown(
+            ["flashcard", "lesson plan", "worksheet", "proofs"],
+            value="flashcard",
+            label="Prompt Type"
+        )
 
+    output = gr.Textbox(label="Generated Output", lines=12, interactive=True)
     btn = gr.Button("Generate")
+
     btn.click(fn=generate_output, inputs=[prompt_type, query], outputs=output)
 
 demo.launch()