Update app.py

app.py

@@ -3,8 +3,8 @@ import gradio as gr
 import google.generativeai as genai
 import requests
 import json
-import time
 import os
+import time
 
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 gemini_model = genai.GenerativeModel("gemini-2.0-flash")
@@ -78,30 +78,22 @@ Question: {question}
     distractors = [line.split(":", 1)[1].strip() for line in lines if line.startswith("Distractor")]
 
     if len(distractors) != 3:
-        return "Error: Could not extract exactly 3 distractors.", gemini_response, ""
+        return gemini_response, "ERROR (Invalid distractor count)"
 
-    claude_llm = OpenRouter("anthropic/claude-3.5-sonnet-20240612", "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")
+    claude_llm = OpenRouter("anthropic/claude-3.5-sonnet-20240612", os.getenv("OPENROUTER_API_KEY"))
     claude_prompt = prepare_claude_ranking_prompt(question, correct, distractors)
     ranking = call_openrouter(claude_llm, claude_prompt)
 
-    output = f"**Question:** {question}\n\n"
-    output += f"**Correct Answer:** {correct}\n"
-    for i, d in enumerate(distractors, 1):
-        output += f"**Distractor {i}:** {d}\n"
-    output += f"\n**Claude Ranking:** {ranking.strip()}"
-
-    return output, gemini_response, claude_prompt
-
+    return gemini_response, ranking.strip()
 
 iface = gr.Interface(
     fn=generate_and_rank,
     inputs=gr.Textbox(label="Enter your MCQ Question"),
     outputs=[
-        gr.Textbox(label="Final Output"),
-        gr.Textbox(label="Raw Gemini Output"),
-        gr.Textbox(label="Claude Prompt")
+        gr.Textbox(label="Generated Answer and Distractors (Gemini)"),
+        gr.Textbox(label="Claude's Ranking")
     ],
     title="Confusing Distractor Generator + Claude Ranking"
 )
 
-iface.launch()
+iface.launch()