Update app.py

app.py

@@ -1,33 +1,34 @@
-# data_loaders/load_datasets.py
-
-from datasets import load_dataset
-
-def load_all_datasets():
-    datasets = {}
-    try:
-        # Linear Algebra
-        datasets["linear_algebra_composed"] = load_dataset("marcov/math_dataset_algebra__linear_2d_composed_promptsource")
-        datasets["linear_algebra_standard"] = load_dataset("marcov/math_dataset_algebra__linear_2d_promptsource")
-
-        # Probability NLI variants
-        datasets["probability_nli_1hop"] = load_dataset("sileod/probability_words_nli", "reasoning_1hop")
-        datasets["probability_nli_2hop"] = load_dataset("sileod/probability_words_nli", "reasoning_2hop")
-        datasets["probability_nli_usnli"] = load_dataset("sileod/probability_words_nli", "usnli")
-
-        # Qwen Math Probability
-        datasets["probability_qwen"] = load_dataset("yxsllgz-uts-org/Math_Consistency-Probability-Qwen2.5-Math-1.5B-Instruct-style1")
-
-        # Supply Chain QA
-        datasets["supply_chain_qa"] = load_dataset("sebdg/supply_chain_q_a")
-
-        # Game Theory
-        datasets["game_theory"] = load_dataset("ebowwa/game-theory-io")
-
-    except Exception as e:
-        print(f"❌ Error loading one or more datasets: {e}")
-
-    print("📊 Loaded datasets:")
-    for key in datasets.keys():
-        print(f" - {key}")
-
-    return datasets
+import os
+import gradio as gr
+from dotenv import load_dotenv
+
+from agents.openai_agent import query_openai
+from data_loaders.load_datasets import load_all_datasets
+from tools.math_tools import run_math_analysis
+from tools.serpapi_tools import search_serpapi
+
+load_dotenv()
+
+# Callbacks
+def openai_response(prompt):
+    return query_openai(prompt)
+
+def run_tool(topic):
+    return run_math_analysis(topic)
+
+def serpapi_search(query):
+    results = search_serpapi(query)
+    return "\n".join([f"{r['title']}: {r['link']}" for r in results])
+
+def load_datasets_button():
+    datasets = load_all_datasets()
+    return "✅ Datasets loaded successfully:\n" + "\n".join(datasets.keys())
+
+# Gradio Interface
+with gr.Blocks() as demo:
+    gr.Markdown("## 🤖 AI Business Agent (Math + Models + Data)")
+    
+    with gr.Tab("1. Ask OpenAI"):
+        q = gr.Textbox(label="Your question to GPT-4")
+        a = gr.Textbox(label="Response", lines=10)
+        q_btn = gr.Button("Ask")