Update app.py

app.py

@@ -12,8 +12,9 @@ from langchain_openai import ChatOpenAI
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 HUGGINGFACEHUB_API_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
-# REPO_ID = "Qwen/Qwen2.5-Coder-7B-Instruct"
 REPO_ID = "meta-llama/Llama-3.1-8B-Instruct"
+PROVIDER_TYPE = "openai"  # "openai" or "huggingface"
+
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
@@ -34,17 +35,26 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     questions_url = f"{api_url}/questions"
     submit_url = f"{api_url}/submit"
 
-    # 1. Instantiate Agent ( modify this part to create your agent)
+    # 1. Instantiate Agent 
     try:
-        llm = HuggingFaceEndpoint(
-            repo_id=REPO_ID,
-            huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
-        )
-        agent = BasicAgent(llm)
+        if PROVIDER_TYPE=="huggingface":
+            llm = HuggingFaceEndpoint(
+                repo_id=REPO_ID,
+                huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
+            )
+            chat = ChatHuggingFace(llm=llm, verbose=True)
+        elif PROVIDER_TYPE=="openai":
+            chat = ChatOpenAI(model="gpt-4o")
+        else:
+            print(f"Provider {PROVIDER_TYPE} not supported.")
+            return f"Provider {PROVIDER_TYPE} not supported", None
+        agent = BasicAgent(chat)
+
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
-    # In the case of an app running as a hugging Face space, this link points toward your codebase ( usefull for others so please keep it public)
+        
+    # In the case of an app running as a hugging Face space, this link points toward your codebase
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
     print(agent_code)
 
@@ -147,14 +157,13 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
 # --- Build Gradio Interface using Blocks ---
 with gr.Blocks() as demo:
-    gr.Markdown("# Basic Agent Evaluation Runner")
+    gr.Markdown("# Agent Evaluation Runner")
     gr.Markdown(
         """
         **Instructions:**
 
-        1.  Please clone this space, then modify the code to define your agent's logic, the tools, the necessary packages, etc ...
-        2.  Log in to your Hugging Face account using the button below. This uses your HF username for submission.
-        3.  Click 'Run Evaluation & Submit All Answers' to fetch questions, run your agent, submit answers, and see the score.
+        1.  Log in to your Hugging Face account using the button below. This uses your HF username for submission.
+        2.  Click 'Run Evaluation & Submit All Answers' to fetch questions, run your agent, submit answers, and see the score.
 
         ---
         **Disclaimers:**