Update app.py

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 import pandas as pd
 
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
-GEMINI_MODEL = "gemini-1.5-pro"
+GEMINI_MODEL = "gemini-2.5-pro"
 GEMINI_API_URL = f"https://generativelanguage.googleapis.com/v1/models/{GEMINI_MODEL}:generateContent"
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
@@ -25,26 +25,24 @@ class GeminiAgent:
             try:
                 response = requests.post(GEMINI_API_URL, headers=headers, params=params, json=payload)
                 if response.status_code == 429:
-                    print(f"⚠️ Попытка {attempt+1}: 429 Too Many Requests — ждём 5 сек...")
+                    print(f"⚠️  Попытка {attempt+1}: Превышен лимит. Ждём 5 сек.")
                     time.sleep(5)
                     continue
                 response.raise_for_status()
-                raw = response.json()
-                answer = raw["candidates"][0]["content"]["parts"][0]["text"]
-                return str(answer).strip()
+                return response.json()["candidates"][0]["content"]["parts"][0]["text"].strip()
             except Exception as e:
-                print(f"❌ Ошибка Gemini API: {e}")
+                print(f"❌ Ошибка: {e}")
                 time.sleep(5)
-        return "Не удалось получить ответ. Попробуйте позже."
+        return "Ошибка при получении ответа от Gemini API"
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
     if not profile:
-        return "⚠️ Пожалуйста, войдите в Hugging Face", None
+        return "Пожалуйста, войдите в Hugging Face", None
 
     username = profile.username
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
-    print(f"👤 Пользователь: {username}")
+    print(f"Пользователь: {username}")
 
     try:
         response = requests.get(f"{DEFAULT_API_URL}/questions", timeout=15)
@@ -57,17 +55,15 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     results_log = []
     answers_payload = []
 
-    for i, item in enumerate(questions_data):
+    for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
         if not task_id or not question_text:
             continue
-        answer = str(agent(question_text)).strip()
+        answer = agent(question_text)
         results_log.append({"Task ID": task_id, "Question": question_text, "Answer": answer})
-        if task_id and answer:
-            answers_payload.append({"task_id": task_id, "submitted_answer": answer})
-        print(f"✅ [{i+1}/{len(questions_data)}] ID: {task_id} | Ответ: {answer[:50]}")
-        time.sleep(3)
+        answers_payload.append({"task_id": task_id, "submitted_answer": answer})
+        time.sleep(3)  # ⏳ Задержка между запросами
 
     try:
         submission_data = {
@@ -75,25 +71,21 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             "agent_code": agent_code,
             "answers": answers_payload
         }
-
-        print("📤 Пример отправки:", submission_data["answers"][:2])  # Для отладки
         submit_response = requests.post(f"{DEFAULT_API_URL}/submit", json=submission_data)
         submit_response.raise_for_status()
         result = submit_response.json()
         score = result.get("score", "N/A")
-        correct = result.get("correct_count", "?")
-        total = result.get("total_attempted", "?")
-        return f"✅ Отправка завершена!\nРезультат: {score}% ({correct}/{total})", pd.DataFrame(results_log)
+        return f"✅ Отправка завершена! Ваш результат: {score}%", pd.DataFrame(results_log)
     except Exception as e:
-        return f"❌ Ошибка при отправке: {e}", pd.DataFrame(results_log)
+        return f"❌ Ошибка при отправке ответов: {e}", pd.DataFrame(results_log)
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## 🤖 Gemini 2.5 Agent")
-    gr.Markdown("Войдите в Hugging Face, затем нажмите кнопку запуска.")
+    gr.Markdown("## Gemini 2.5 Agent 🤖")
+    gr.Markdown("Войдите в Hugging Face и нажмите кнопку для запуска агента.")
     gr.LoginButton()
-    run_btn = gr.Button("🚀 Запустить агента")
-    status = gr.Textbox(label="Статус выполнения")
+    run_btn = gr.Button("▶️ Запустить агента")
+    status = gr.Textbox(label="Статус")
     table = gr.DataFrame(label="Результаты")
     run_btn.click(fn=run_and_submit_all, outputs=[status, table])