Update app.py

app.py

@@ -5,101 +5,99 @@ import os  # For environment variables
 # Initialize the Hugging Face Inference Client
 client = InferenceClient()
 
-# Functions for solving problems, generating hints, and more
-def solve_math_problem(problem, difficulty):
-    prompt = os.getenv("PROMPT_SOLVE").format(problem=problem, difficulty=difficulty)
-    response = client.chat.completions.create(
-        model="Qwen/QwQ-32B-Preview",
+# Generic function to generate AI response with streaming
+def generate_response(prompt_template, **kwargs):
+    # Format the prompt with provided arguments
+    prompt = os.getenv(prompt_template).format(**kwargs)
+    # Stream the response from the model
+    stream = client.chat.completions.create(
+        model="Qwen/Qwen2.5-Math-72B-Instruct",
         messages=[{"role": "user", "content": prompt}],
         temperature=0.7,
         max_tokens=1024,
-        top_p=0.8
+        top_p=0.8,
+        stream=True
     )
-    return response.choices[0].message["content"]
+    # Stream chunks as they are generated
+    response = ""
+    for chunk in stream:
+        response += chunk.choices[0].delta.content
+        yield response
 
-def generate_hint(problem, difficulty):
-    prompt = os.getenv("PROMPT_HINT").format(problem=problem, difficulty=difficulty)
-    response = client.chat.completions.create(
-        model="Qwen/QwQ-32B-Preview",
-        messages=[{"role": "user", "content": prompt}],
-        temperature=0.7,
-        max_tokens=512,
-        top_p=0.8
-    )
-    return response.choices[0].message["content"]
-
-def verify_solution(problem, solution):
-    prompt = os.getenv("PROMPT_VERIFY").format(problem=problem, solution=solution)
-    response = client.chat.completions.create(
-        model="Qwen/QwQ-32B-Preview",
-        messages=[{"role": "user", "content": prompt}],
-        temperature=0.7,
-        max_tokens=512,
-        top_p=0.8
-    )
-    return response.choices[0].message["content"]
-
-def generate_practice_question(topic, difficulty):
-    prompt = os.getenv("PROMPT_GENERATE").format(topic=topic, difficulty=difficulty)
-    response = client.chat.completions.create(
-        model="Qwen/QwQ-32B-Preview",
-        messages=[{"role": "user", "content": prompt}],
-        temperature=0.7,
-        max_tokens=512,
-        top_p=0.8
-    )
-    return response.choices[0].message["content"]
-
-def explain_concept(problem, difficulty):
-    prompt = os.getenv("PROMPT_EXPLAIN").format(problem=problem, difficulty=difficulty)
-    response = client.chat.completions.create(
-        model="Qwen/QwQ-32B-Preview",
-        messages=[{"role": "user", "content": prompt}],
-        temperature=0.7,
-        max_tokens=512,
-        top_p=0.8
-    )
-    return response.choices[0].message["content"]
-
-# Create Gradio interface
+# Gradio app interface
 with gr.Blocks() as app:
     gr.Markdown("## Mathematical Insight Tutor")
     gr.Markdown("An advanced AI-powered tutor to help you master math concepts.")
 
-    with gr.Tab("Solve a Problem"):
-        problem_input = gr.Textbox(label="Enter Math Problem", placeholder="e.g., Solve for x: 2x + 5 = 15")
-        difficulty_level = gr.Dropdown(choices=["Beginner", "Intermediate", "Advanced"], label="Difficulty Level")
-        solve_button = gr.Button("Solve")
-        solution_output = gr.Markdown()
-        solve_button.click(solve_math_problem, inputs=[problem_input, difficulty_level], outputs=solution_output)
+    # Function to create a reusable tab with inputs and outputs
+    def create_tab(tab_name, prompt_template, inputs):
+        with gr.Tab(tab_name):
+            input_fields = []
+            # Create input components dynamically
+            for inp in inputs:
+                if inp["type"] == "textbox":
+                    input_fields.append(
+                        gr.Textbox(lines=inp.get("lines", 1), label=inp["label"], placeholder=inp["placeholder"])
+                    )
+                elif inp["type"] == "dropdown":
+                    input_fields.append(
+                        gr.Dropdown(choices=inp["choices"], label=inp["label"], value=inp.get("default", ""))
+                    )
+            # Button and output
+            button = gr.Button(f"{tab_name} Execute")
+            output = gr.Markdown()
+            # Link button to the response generator
+            button.click(
+                fn=lambda *args: generate_response(prompt_template, **dict(zip([inp["key"] for inp in inputs], args))),
+                inputs=input_fields,
+                outputs=output
+            )
 
-    with gr.Tab("Generate a Hint"):
-        hint_problem_input = gr.Textbox(label="Enter Math Problem for Hint", placeholder="e.g., Solve for x: 2x + 5 = 15")
-        hint_difficulty = gr.Dropdown(choices=["Beginner", "Intermediate", "Advanced"], label="Difficulty Level")
-        hint_button = gr.Button("Generate Hint")
-        hint_output = gr.Markdown()
-        hint_button.click(generate_hint, inputs=[hint_problem_input, hint_difficulty], outputs=hint_output)
+    # Tabs for various functionalities
+    create_tab(
+        "Solve a Problem",
+        "PROMPT_SOLVE",
+        [
+            {"key": "problem", "type": "textbox", "label": "Enter Math Problem", "placeholder": "e.g., Solve for x: 2x + 5 = 15"},
+            {"key": "difficulty", "type": "dropdown", "label": "Difficulty Level", "choices": ["Beginner", "Intermediate", "Advanced"]}
+        ]
+    )
 
-    with gr.Tab("Verify Solution"):
-        verify_problem_input = gr.Textbox(label="Enter Math Problem", placeholder="e.g., Solve for x: 2x + 5 = 15")
-        solution_input = gr.Textbox(label="Enter Your Solution", placeholder="e.g., x = 5")
-        verify_button = gr.Button("Verify Solution")
-        verify_output = gr.Markdown()
-        verify_button.click(verify_solution, inputs=[verify_problem_input, solution_input], outputs=verify_output)
+    create_tab(
+        "Generate a Hint",
+        "PROMPT_HINT",
+        [
+            {"key": "problem", "type": "textbox", "label": "Enter Math Problem for Hint", "placeholder": "e.g., Solve for x: 2x + 5 = 15"},
+            {"key": "difficulty", "type": "dropdown", "label": "Difficulty Level", "choices": ["Beginner", "Intermediate", "Advanced"]}
+        ]
+    )
 
-    with gr.Tab("Generate Practice Question"):
-        topic_input = gr.Textbox(label="Enter Math Topic", placeholder="e.g., Algebra, Calculus")
-        practice_difficulty = gr.Dropdown(choices=["Beginner", "Intermediate", "Advanced"], label="Difficulty Level")
-        generate_button = gr.Button("Generate Question")
-        practice_output = gr.Markdown()
-        generate_button.click(generate_practice_question, inputs=[topic_input, practice_difficulty], outputs=practice_output)
+    create_tab(
+        "Verify Solution",
+        "PROMPT_VERIFY",
+        [
+            {"key": "problem", "type": "textbox", "label": "Enter Math Problem", "placeholder": "e.g., Solve for x: 2x + 5 = 15"},
+            {"key": "solution", "type": "textbox", "label": "Enter Your Solution", "placeholder": "e.g., x = 5"}
+        ]
+    )
 
-    with gr.Tab("Explain Concept"):
-        explain_problem_input = gr.Textbox(label="Enter Math Problem", placeholder="e.g., Solve for x: 2x + 5 = 15")
-        explain_difficulty = gr.Dropdown(choices=["Beginner", "Intermediate", "Advanced"], label="Difficulty Level")
-        explain_button = gr.Button("Explain Concept")
-        explain_output = gr.Markdown()
-        explain_button.click(explain_concept, inputs=[explain_problem_input, explain_difficulty], outputs=explain_output)
+    create_tab(
+        "Generate Practice Question",
+        "PROMPT_GENERATE",
+        [
+            {"key": "topic", "type": "textbox", "label": "Enter Math Topic", "placeholder": "e.g., Algebra, Calculus"},
+            {"key": "difficulty", "type": "dropdown", "label": "Difficulty Level", "choices": ["Beginner", "Intermediate", "Advanced"]}
+        ]
+    )
+
+    create_tab(
+        "Explain Concept",
+        "PROMPT_EXPLAIN",
+        [
+            {"key": "problem", "type": "textbox", "label": "Enter Math Problem", "placeholder": "e.g., Solve for x: 2x + 5 = 15"},
+            {"key": "difficulty", "type": "dropdown", "label": "Difficulty Level", "choices": ["Beginner", "Intermediate", "Advanced"]}
+        ]
+    )
 
 # Launch the app
 app.launch(debug=True)