Update app.py

app.py

@@ -5,22 +5,25 @@ from huggingface_hub import InferenceClient
 client = InferenceClient()
 
 # Function to stream content for Math, STEM, and Code Generation
-def generate_stream(selected_topic, input_text):
+def generate_stream(selected_topic, subtopic, input_text, examples_count):
     """
     Generates dynamic lessons, solutions, or code snippets based on the selected topic.
 
     Args:
         selected_topic (str): The selected subject (e.g., Math, STEM, or Code Generation).
+        subtopic (str): Specific subtopic or category for more focused output.
         input_text (str): Additional input for contextual content generation.
+        examples_count (int): Number of examples to generate.
 
     Yields:
         str: Incremental output content.
     """
     # Create a topic-specific prompt
     prompt = (
-        f"Generate a {selected_topic.lower()} lesson, problem, or example based on the following input: {input_text}"
+        f"Generate {examples_count} detailed {selected_topic.lower()} examples, lessons, or problems "
+        f"focused on {subtopic}. Input context: {input_text}"
         if input_text.strip() else
-        f"Generate a beginner-level {selected_topic.lower()} lesson with examples."
+        f"Generate {examples_count} beginner-level {selected_topic.lower()} lessons or examples on {subtopic}."
     )
     messages = [{"role": "user", "content": prompt}]
 
@@ -46,10 +49,10 @@ def generate_stream(selected_topic, input_text):
 # Create the Gradio interface
 with gr.Blocks() as app:
     # App Title and Instructions
-    gr.Markdown("## 🎓 STEM Learning and Code Generator")
+    gr.Markdown("## 🎓 Enhanced STEM Learning and Code Generator")
     gr.Markdown(
-        "Get dynamic lessons, problem-solving examples, or code snippets for Math, STEM, "
-        "or Computer Science. Select a topic and get started!"
+        "Generate tailored lessons, problem-solving examples, or code snippets for Math, STEM, "
+        "or Computer Science. Select a topic, subtopic, and customize your experience!"
     )
 
     with gr.Row():
@@ -60,28 +63,53 @@ with gr.Blocks() as app:
                 label="Select a Topic",
                 value="Math"  # Default selection
             )
+            subtopic = gr.Textbox(
+                lines=1,
+                label="Subtopic",
+                placeholder="Specify a subtopic (e.g., Algebra, Physics, Data Structures)."
+            )
             input_text = gr.Textbox(
                 lines=2,
-                label="Optional Input",
+                label="Context or Additional Input",
                 placeholder="Provide additional context (e.g., 'Explain calculus basics' or 'Generate Python code for sorting')."
             )
+            examples_count = gr.Slider(
+                minimum=1,
+                maximum=5,
+                value=1,
+                step=1,
+                label="Number of Examples"
+            )
             generate_button = gr.Button("Generate Content")
 
         # Output Section
         with gr.Column():
             gr.Markdown("### Generated Content")
-            output_stream = gr.TextArea(
+            output_stream = gr.Textbox(
+                lines=20,
                 label="Output",
-                placeholder="Generated content will appear here...",
                 interactive=False
             )
+            export_button = gr.Button("Export Code (if applicable)")
 
     # Link the generate button to the streaming function
     generate_button.click(
         fn=generate_stream,
-        inputs=[selected_topic, input_text],
+        inputs=[selected_topic, subtopic, input_text, examples_count],
         outputs=output_stream
     )
 
+    # Export functionality for code snippets
+    def export_code(content):
+        with open("generated_code.py", "w") as file:
+            file.write(content)
+        return "Code exported successfully to generated_code.py!"
+
+    export_button.click(
+        fn=export_code,
+        inputs=[output_stream],
+        outputs=[output_stream]
+    )
+
 # Launch the Gradio app
 app.launch()