Modified the Gradio interface

.env.example

@@ -13,4 +13,9 @@ CHESS_ENGINE_PATH=stock-fish-engine-location-here
 # Configure this if you want to enable observability with LangSmith
 LANGSMITH_API_KEY=your-langsmith-key
 LANGSMITH_TRACING=false
-LANGSMITH_PROJECT=gaia_agent
+LANGSMITH_PROJECT=gaia_agent
+
+# Flag indicating whether the agent will work in submission mode or not.
+# If submission mode is on the agent will retrieve 20 questions and submit their answers
+# If submission mode is off the agent will accept questions from the user
+SUBMISSION_MODE_ON=False

README.md

@@ -123,7 +123,8 @@ to the `stockfish` executable, otherwise the `stockfish`installation is automati
 
 The `LANGSMITH_*` properties need to be configured only if you want to enable observability with LangSmith.
 
-
+The `SUBMISSION_MODE_ON` flag indicates whether the application will run in submission mode (when the 20 questions are fetched and the answers are
+submitted for agent evaluation) or not (the agent accepts a question and an attachment).
 
 ## References 📚
 The math tool implementation was inspired from this repo https://github.com/langchain-ai/open_deep_research

app.py

@@ -1,11 +1,15 @@
 import os
+import shutil
+import tempfile
 import time
+import uuid
 from pathlib import Path
 
 import gradio as gr
 import pandas as pd
 import requests
 
+from config.settings import config
 from core.agent import GaiaAgent, Attachment
 from utils.cache_answers import AnswersCache
 from utils.dependencies_checker import check_dependencies
@@ -122,7 +126,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
                     print(f"Agent received an attachment : {attachment.file_path}...")
                 submitted_answer = agent(question_text, attachment)
                 print(f"Agent returning fixed answer: {submitted_answer}")
-                #sleep in
+                # sleep in
                 time.sleep(30)
                 cache.set(task_id, submitted_answer)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
@@ -140,7 +144,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     status_update = f"Agent finished. Submitting {len(answers_payload)} answers for user '{username}'..."
     print(status_update)
 
-    #5. Submit
+    # 5. Submit
     print(f"Submitting {len(answers_payload)} answers to: {submit_url}")
     try:
         response = requests.post(submit_url, json=submission_data, timeout=60)
@@ -185,7 +189,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
 
 # --- Build Gradio Interface using Blocks ---
-with gr.Blocks() as demo:
+with gr.Blocks() as demo_submit:
     gr.Markdown("# Basic Agent Evaluation Runner")
     gr.Markdown(
         """
@@ -215,6 +219,75 @@ with gr.Blocks() as demo:
         outputs=[status_output, results_table]
     )
 
+
+def process_input(question: str, file: gr.File):
+    """
+    Process the user's question and attached file
+    """
+    if not question:
+        return "Please enter a question."
+
+    # Extract file information
+    attachment = None
+    if file is not None:
+        print(f"Received file {file.name} ")
+
+        # Save to disk
+        task_id = uuid.uuid4()
+        file_name = Path(file.name).name
+        file_path = Path("uploads") / f"{task_id}" / f"{file_name}"
+
+        # # Create parent directories if they don't exist
+        file_path.parent.mkdir(parents=True, exist_ok=True)
+        shutil.copy(file, file_path)
+
+        content = file_path.read_bytes()
+        attachment = Attachment(content, file_path.as_posix())
+
+    response = agent(question, attachment)
+
+    return response
+
+
+with gr.Blocks(title="🐉 GAIA Agent Demo", theme=gr.themes.Ocean()) as demo:
+    gr.Markdown("# 🐉 GAIA Agent")
+    gr.Markdown("Ask me a complex question")
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            question_input = gr.Textbox(
+                label="Your Question",
+                placeholder="Type your question here",
+                lines=5
+            )
+
+            file_input = gr.File(
+                label="Attach File",
+                file_types=[
+                    ".txt", ".pdf", ".png", ".jpg", ".jpeg",
+                    ".csv", ".py", ".mp3", ".xslx"
+                ]
+            )
+
+            submit_btn = gr.Button("Submit", variant="primary")
+
+        with gr.Column(scale=2):
+            output = gr.Textbox(label="Response", lines=10)
+
+    # Set up the submission action
+    submit_btn.click(
+        fn=process_input,
+        inputs=[question_input, file_input],
+        outputs=output
+    )
+
+    # Also process when file is uploaded (optional)
+    file_input.upload(
+        fn=process_input,
+        inputs=[question_input, file_input],
+        outputs=output
+    )
+
 if __name__ == "__main__":
     print("\n" + "-" * 30 + " App Starting " + "-" * 30)
     # Check for SPACE_HOST and SPACE_ID at startup for information
@@ -237,6 +310,10 @@ if __name__ == "__main__":
     print("-" * (60 + len(" App Starting ")) + "\n")
     print("-" * (60 + len(" Check dependencies ")) + "\n")
     check_dependencies()
+    agent = GaiaAgent()
 
-    print("Launching Gradio Interface for Basic Agent Evaluation...")
-    demo.launch(debug=True, share=False)
+    if config.submission_mode_on:
+        print("Launching Gradio Interface for Basic Agent Evaluation...")
+        demo_submit.launch(debug=True, share=False)
+    else:
+        demo.launch(debug=True, share=False)

config/settings.py

@@ -25,6 +25,8 @@ class AgentConfig(BaseSettings):
     project_root: Path = Path(__file__).parent.parent
     prompts_location: Path = project_root / "config" / "prompts.yaml"
 
+    submission_mode_on: bool = False
+
     class Config:
         env_file = ".env"
         case_sensitive = False

core/agent.py

@@ -8,8 +8,7 @@ from langgraph.prebuilt import tools_condition
 
 from core.messages import Attachment
 from core.state import State
-from nodes.nodes import assistant, optimize_memory, response_processing, pre_processor
-from tools.tavily_tools import web_search_tools
+from nodes.nodes import assistant, optimize_memory, response_processing, pre_processor, agent_tools
 
 
 class GaiaAgent:
@@ -22,7 +21,7 @@ class GaiaAgent:
         # Define nodes: these do the work
         builder.add_node("pre_processor", pre_processor)
         builder.add_node("assistant", assistant)
-        builder.add_node("tools", ToolNode(web_search_tools))
+        builder.add_node("tools", ToolNode(agent_tools))
         builder.add_node("optimize_memory", optimize_memory)
         builder.add_node("response_processing", response_processing)