Update app.py

app.py

@@ -1,12 +1,12 @@
 import gradio as gr
-import genai  # Assuming this is the correct library for Gemini API
-import sympy as sp
+import genai  # Assuming you are using a `genai` library for Gemini
 
-# Set up the Gemini model
-model = genai.GenerativeModel("gemini-1.5-flash")
+# Set up the Gemini generative model using your API key
+model = genai.GenerativeModel("gemini-1.5-flash", api_key="AIzaSyCjAahw7YvVFWSYHWRaGwZ9cA9emobfNok")
 
-# Function for algebraic simplification
+# Define the functions Gemini will call
 def algebra_simplification(expression):
+    import sympy as sp
     try:
         expr = sp.sympify(expression)
         simplified_expr = sp.simplify(expr)
@@ -14,8 +14,8 @@ def algebra_simplification(expression):
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Function for solving algebraic equations
 def algebra_solve(equation):
+    import sympy as sp
     try:
         eq = sp.sympify(equation)
         variables = list(eq.free_symbols)
@@ -24,32 +24,42 @@ def algebra_solve(equation):
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Function to handle the chat with Gemini
-def process_math_problem(math_problem):
-    chat = model.start_chat(
-        history=[
-            {"role": "user", "parts": "Hello"},
-            {"role": "model", "parts": "Great to meet you. What would you like to know?"},
-        ]
-    )
+# Chat model and function calling logic
+def gemini_chat(math_problem):
+    chat = model.start_chat(history=[])
     
-    # Send the math problem to Gemini
-    response = chat.send_message(math_problem)
+    # Gemini model should figure out whether to call the appropriate function
+    response = chat.send_message(math_problem, functions=[
+        {
+            "name": "algebra_simplification",
+            "parameters": {
+                "expression": math_problem
+            },
+            "fn": algebra_simplification
+        },
+        {
+            "name": "algebra_solve",
+            "parameters": {
+                "equation": math_problem
+            },
+            "fn": algebra_solve
+        }
+    ], function_call="auto")  # Set function_call to 'auto'
     
-    # Return Gemini's response
     return response.text
 
-# Gradio interface
+# Gradio Interface
+def process_math_problem(math_problem):
+    # Use Gemini's auto function calling to process the math problem
+    return gemini_chat(math_problem)
+
 with gr.Blocks() as demo:
     gr.Markdown("# Gemini Algebra Solver")
     
-    # Single input field for math problems
     input_problem = gr.Textbox(label="Enter a Math Problem", placeholder="e.g., Simplify (x^2 - 4)/(x - 2) or Solve x^2 - 4 = 0")
     output_solution = gr.Textbox(label="Solution")
     
-    # Button to trigger the solution process
     solve_button = gr.Button("Solve")
     solve_button.click(fn=process_math_problem, inputs=input_problem, outputs=output_solution)
 
-# Launch the Gradio interface
 demo.launch()