Update app.py with full Gaia LlamaIndex agent

app.py

@@ -1,20 +1,31 @@
-import gradio as gr
 import os
 import sys
 import json
+import traceback
 from typing import List, Dict
 
-# Add the current directory to Python path
+import gradio as gr
+import nltk
+
+# --- Environment variable setup to fix permission issues in Spaces or restricted envs ---
+os.environ["NLTK_DATA"] = "/tmp/nltk_data"
+os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface_cache"
+
+# Download required NLTK data upfront
+nltk.download('punkt', download_dir=os.environ["NLTK_DATA"])
+
+# Add current directory to path for local imports
 sys.path.append(os.path.dirname(os.path.abspath(__file__)))
 
-# ✅ Ensure GaiaAPI is imported at the top level
+# Import GaiaAPI early
 from utils.gaia_api import GaiaAPI
 
-# Initialize variables
+# Initialize global agent state
 AGENT_READY = False
 agent = None
 
-# Import other agent modules
+# Import agent-related modules and initialize
 try:
     print("Importing modules...")
     from agent.local_llm import LocalLLM
@@ -24,7 +35,6 @@ try:
 
     print("All imports successful!")
 
-    # Initialize components
     print("Initializing Local LLM...")
     local_llm = LocalLLM()
     llm = local_llm.get_llm()
@@ -44,11 +54,11 @@ try:
 
 except Exception as e:
     print(f"Failed to initialize agent: {str(e)}")
-    import traceback
     traceback.print_exc()
     AGENT_READY = False
     agent = None
 
+
 def process_single_question(question_text: str) -> str:
     """Process a single GAIA question through the agent"""
     if not AGENT_READY:
@@ -59,14 +69,14 @@ def process_single_question(question_text: str) -> str:
 
     try:
         enhanced_prompt = f"""
-        Answer the following question directly and concisely. Do not include "FINAL ANSWER" or any other prefixes in your response. Just provide the answer.
+Answer the following question directly and concisely. Do not include "FINAL ANSWER" or any other prefixes in your response. Just provide the answer.
 
-        Question: {question_text}
-        """
+Question: {question_text}
+"""
         response = agent.query(enhanced_prompt)
         answer = str(response).strip()
 
-        # Remove common prefixes
+        # Remove common prefixes from the answer
         for prefix in ["FINAL ANSWER:", "Answer:", "The answer is:", "Final answer:"]:
             if answer.startswith(prefix):
                 answer = answer[len(prefix):].strip()
@@ -74,9 +84,9 @@ def process_single_question(question_text: str) -> str:
         return answer
 
     except Exception as e:
-        import traceback
         return f"❌ Error: {str(e)}\n\n{traceback.format_exc()}"
 
+
 def process_all_questions() -> str:
     """Process all GAIA questions and prepare answers for submission"""
     if not AGENT_READY:
@@ -87,7 +97,7 @@ def process_all_questions() -> str:
         processed_answers = []
 
         for i, question in enumerate(questions):
-            print(f"Processing question {i+1}/{len(questions)}: {question['task_id']}")
+            print(f"Processing question {i + 1}/{len(questions)}: {question['task_id']}")
             answer = process_single_question(question['question'])
             processed_answers.append({
                 "task_id": question['task_id'],
@@ -105,9 +115,9 @@ def process_all_questions() -> str:
         return summary
 
     except Exception as e:
-        import traceback
         return f"❌ Error: {str(e)}\n\n{traceback.format_exc()}"
 
+
 def submit_to_gaia(username: str, code_url: str) -> str:
     """Submit answers to GAIA benchmark"""
     if not AGENT_READY:
@@ -131,6 +141,7 @@ def submit_to_gaia(username: str, code_url: str) -> str:
     except Exception as e:
         return f"❌ Submission error: {str(e)}"
 
+
 def get_sample_question() -> str:
     """Load a sample question for testing"""
     try:
@@ -139,15 +150,16 @@ def get_sample_question() -> str:
     except Exception as e:
         return f"Error loading sample question: {str(e)}"
 
+
 # ---------- Gradio UI ----------
 with gr.Blocks(title="🦙 GAIA LlamaIndex Agent") as demo:
     gr.Markdown(f"""
-    # 🦙 GAIA Benchmark Agent with LlamaIndex
-    
-    This agent uses LlamaIndex with a local LLM to tackle GAIA benchmark questions.
-    
-    **Status:** {"✅ Ready" if AGENT_READY else "❌ Not Ready"}
-    """)
+# 🦙 GAIA Benchmark Agent with LlamaIndex
+
+This agent uses LlamaIndex with a local LLM to tackle GAIA benchmark questions.
+
+**Status:** {"✅ Ready" if AGENT_READY else "❌ Not Ready"}
+""")
 
     with gr.Tab("🔬 Test Single Question"):
         gr.Markdown("Test the agent with individual questions")
@@ -183,12 +195,12 @@ with gr.Blocks(title="🦙 GAIA LlamaIndex Agent") as demo:
 
     with gr.Tab("🏆 Submit to GAIA"):
         gr.Markdown("""
-        Submit your processed answers to the GAIA benchmark for official scoring.
+Submit your processed answers to the GAIA benchmark for official scoring.
 
-        **Requirements:**
-        1. Your Hugging Face username
-        2. Link to your Space code (e.g., https://huggingface.co/spaces/your-username/gaia-agent)
-        """)
+**Requirements:**
+1. Your Hugging Face username
+2. Link to your Space code (e.g., https://huggingface.co/spaces/your-username/gaia-agent)
+""")
 
         with gr.Row():
             with gr.Column():
@@ -203,17 +215,17 @@ with gr.Blocks(title="🦙 GAIA LlamaIndex Agent") as demo:
 
     with gr.Tab("ℹ️ Info & Debug"):
         gr.Markdown(f"""
-        ## About This Agent
-        
-        This agent uses:
-        - **LlamaIndex** (ReAct Agent + Tools)
-        - **Local LLM** (e.g., DialoGPT or fallback GPT2)
-        - **GAIA Tools** (question fetch, file reader, math, etc.)
-        
-        ## Current Status
-        - Agent Ready: {"✅ Yes" if AGENT_READY else "❌ No"}
-        - Tools Loaded: {len(gaia_tools) if 'gaia_tools' in globals() else 0}
-        """)
+## About This Agent
+
+This agent uses:
+- **LlamaIndex** (ReAct Agent + Tools)
+- **Local LLM** (e.g., DialoGPT or fallback GPT2)
+- **GAIA Tools** (question fetch, file reader, math, etc.)
+
+## Current Status
+- Agent Ready: {"✅ Yes" if AGENT_READY else "❌ No"}
+- Tools Loaded: {len(gaia_tools) if 'gaia_tools' in globals() else 0}
+""")
 
 if __name__ == "__main__":
-    demo.launch(show_error=True)
+    demo.launch(server_name="0.0.0.0", server_port=7860, show_error=True)