Update app.py

app.py

@@ -1,49 +1,24 @@
-# Import necessary libraries
 import streamlit as st
-import autogen
-from autogen.agentchat.contrib.math_user_proxy_agent import MathUserProxyAgent
+import subprocess
+import json
 
-# Function to run the query
-def run_query(math_problem, api_key):
-    config_list = [
-        {
-            'model': 'gpt-3.5-turbo',
-            'api_key': api_key,
-        },
-    ]
-
-    autogen.ChatCompletion.start_logging()
-
-    assistant = autogen.AssistantAgent(
-        name="assistant",
-        system_message="You are a helpful assistant.",
-        llm_config={
-            "request_timeout": 600,
-            "seed": 42,
-            "config_list": config_list,
-        }
-    )
-
-    mathproxyagent = MathUserProxyAgent(
-        name="mathproxyagent",
-        human_input_mode="NEVER",
-        code_execution_config={"use_docker": False},
-    )
-
-    return mathproxyagent.initiate_chat(assistant, problem=math_problem)
-   
-
-# Streamlit app
 st.title('Math Problem Solver')
 
-# Input fields for API key and math problem
 api_key = st.text_input("Enter your API Key:")
 math_problem = st.text_input("Enter your math problem:")
 
-# When the button is pressed, call run_query and display the result
 if st.button('Solve'):
     if api_key and math_problem:
-        result = run_query(math_problem, api_key)
-        st.write(result)
+        input_data = {
+            "math_problem": math_problem,
+            "api_key": api_key
+        }
+
+        result = subprocess.check_output(
+            ["python", "query_solver.py"],
+            input=json.dumps(input_data),
+            text=True
+        )
+        st.write(result.strip())
     else:
-        st.write("Please provide both API key and math problem.")
+        st.write("Please provide both API key and math problem.")