Update app.py

app.py

@@ -1,51 +1,94 @@
 import gradio as gr
+import requests
 import sympy as sp
 
 # Function for algebraic simplification
 def algebra_simplification(expression):
     try:
-        # Ensure that the input expression is sympified correctly
         expr = sp.sympify(expression)
-        
-        # Check if the expression is actually an algebraic expression
-        if isinstance(expr, (sp.Basic, sp.Expr)):
-            simplified_expr = sp.simplify(expr)  # Simplify the expression
-            return str(simplified_expr)
-        else:
-            return "Input is not a valid algebraic expression."
+        simplified_expr = sp.simplify(expr)
+        return str(simplified_expr)
     except Exception as e:
         return f"Error: {str(e)}"
 
 # Function for solving algebraic equations
 def algebra_solve(equation):
     try:
-        # Ensure that the input equation is sympified correctly
         eq = sp.sympify(equation)
-        
-        # Extract variables from the equation
         variables = list(eq.free_symbols)
-        
-        # Solve the equation
         solutions = sp.solve(eq, *variables)
         return str(solutions)
     except Exception as e:
         return f"Error: {str(e)}"
 
+# Gemini API function calling
+def gemini_api_call(math_problem):
+    url = "https://api.gemini.ai/v1/function_call"
+    headers = {
+        "Authorization": "Bearer YOUR_API_KEY",
+        "Content-Type": "application/json",
+    }
+    
+    # Define function descriptors for Gemini's function calling system
+    functions = [
+        {
+            "name": "algebra_simplification",
+            "description": "Simplifies an algebraic expression",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "expression": {
+                        "type": "string",
+                        "description": "The algebraic expression to simplify"
+                    }
+                },
+                "required": ["expression"]
+            }
+        },
+        {
+            "name": "algebra_solve",
+            "description": "Solves an algebraic equation",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "equation": {
+                        "type": "string",
+                        "description": "The algebraic equation to solve"
+                    }
+                },
+                "required": ["equation"]
+            }
+        }
+    ]
+    
+    # Payload for the Gemini API request
+    data = {
+        "model": "gemini-1.5-flash",
+        "prompt": math_problem,
+        "functions": functions,
+        "function_call": "auto"  # Automatically choose the function to call
+    }
+
+    response = requests.post(url, headers=headers, json=data)
+    
+    if response.status_code == 200:
+        result = response.json()
+        return result['output']  # Adjust according to API response structure
+    else:
+        return f"Error: {response.status_code} - {response.text}"
+
 # Gradio interface
+def process_math_problem(math_problem):
+    # Call Gemini API and return the result
+    return gemini_api_call(math_problem)
+
 with gr.Blocks() as demo:
-    gr.Markdown("# Algebra Tool: Simplify and Solve")
+    gr.Markdown("# Gemini Algebra Solver")
+
+    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")
     
-    with gr.Tab("Simplify"):
-        input_expression = gr.Textbox(label="Enter an Expression", placeholder="e.g., (x^2 - 4)/(x - 2)")
-        output_simplification = gr.Textbox(label="Simplified Expression")
-        simplify_button = gr.Button("Simplify")
-        simplify_button.click(fn=algebra_simplification, inputs=input_expression, outputs=output_simplification)
-
-    with gr.Tab("Solve"):
-        input_equation = gr.Textbox(label="Enter an Equation", placeholder="e.g., x^2 - 4 = 0")
-        output_solution = gr.Textbox(label="Solution")
-        solve_button = gr.Button("Solve")
-        solve_button.click(fn=algebra_solve, inputs=input_equation, outputs=output_solution)
-
-# Launch the Gradio interface
+    solve_button = gr.Button("Solve")
+    solve_button.click(fn=process_math_problem, inputs=input_problem, outputs=output_solution)
+
 demo.launch()