Update app.py

app.py

@@ -18,7 +18,7 @@ import gradio as gr
 import requests
 import json
 
-OR_API_KEY = "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"
+OR_API_KEY = os.getenv("OR_API_KEY")
 HEADERS = {
     "Authorization": f"Bearer {OR_API_KEY}",
     "Content-Type": "application/json",
@@ -26,36 +26,15 @@ HEADERS = {
     "X-Title": "<YOUR_SITE_NAME>", # Optional. Site title for rankings on openrouter.ai.
 }
 
-def get_concise_answer(prompt):
-  response = requests.post(
-    url="https://openrouter.ai/api/v1/chat/completions",
-    headers=HEADERS,
-    data=json.dumps({
-      "model": "deepseek/deepseek-chat-v3-0324:free",
-      "messages": [
-        {
-          "role": "system",
-          "content": "you are a helpful assistant that gives concise response. For example you don't add prefixes or the reflection to the answer message. For instance if the user asks you how many albums and the response is 3 just type 3 only"
-        },
-        {
-          "role": "user",
-          "content": prompt
-        }
-      ],
-    })
-  )
-
-  data = response.json()
-  model_response = data['choices'][0]['message']['content']
-  return model_response
 
+'''
 # Example usage:
 prompt1 = "Hello, what is the cultural capital of morocco?"
 print(get_concise_answer(prompt1))
 
 prompt2 = "how many albums does maher zain have?"
 print(get_concise_answer(prompt2))
-
+'''
 
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -183,6 +162,29 @@ agent = workflow.compile()
 
 # 6. Run Function
 
+def get_concise_answer(prompt):
+  response = requests.post(
+    url="https://openrouter.ai/api/v1/chat/completions",
+    headers=HEADERS,
+    data=json.dumps({
+      "model": "deepseek/deepseek-chat-v3-0324:free",
+      "messages": [
+        {
+          "role": "system",
+          "content": "you are a helpful assistant that gives concise response. For example you don't add prefixes or the reflection to the answer message. For instance if the user asks you how many albums and the response is 3 just type 3 only"
+        },
+        {
+          "role": "user",
+          "content": prompt
+        }
+      ],
+    })
+  )
+
+  data = response.json()
+  model_response = data['choices'][0]['message']['content']
+  return model_response
+
 def run_groq_agent(query: str):
     response = agent.invoke({
         "messages": [HumanMessage(content=query)],
@@ -261,7 +263,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = run_groq_agent(question_text)
+            submitted_answer = get_concise_answer(question_text)
+            #submitted_answer = run_groq_agent(question_text)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e: