Update app.py

app.py

@@ -85,8 +85,20 @@ GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
 
 if GEMINI_API_KEY:
     genai.configure(api_key=GEMINI_API_KEY)
-    gemini_model = genai.GenerativeModel('gemini-1.5-flash')
-    print("✓ Gemini API initialized successfully!")
+    # Use the correct model name for the stable API
+    try:
+        # Try gemini-pro first (most stable)
+        gemini_model = genai.GenerativeModel('gemini-pro')
+        print("✓ Gemini API initialized successfully with gemini-pro!")
+    except Exception as e:
+        print(f"Failed to load gemini-pro: {e}")
+        try:
+            # Fallback to gemini-1.5-flash-latest
+            gemini_model = genai.GenerativeModel('gemini-1.5-flash-latest')
+            print("✓ Gemini API initialized successfully with gemini-1.5-flash-latest!")
+        except Exception as e2:
+            print(f"Failed to load gemini-1.5-flash-latest: {e2}")
+            gemini_model = None
 else:
     gemini_model = None
     print("⚠️ Warning: GEMINI_API_KEY not found in environment variables")
@@ -245,11 +257,36 @@ def generate_feedback_from_prompt(prompt_input: str) -> str:
             top_p=0.9,
         )
         
+        # Add safety settings to prevent blocking
+        safety_settings = [
+            {
+                "category": "HARM_CATEGORY_HARASSMENT",
+                "threshold": "BLOCK_NONE"
+            },
+            {
+                "category": "HARM_CATEGORY_HATE_SPEECH",
+                "threshold": "BLOCK_NONE"
+            },
+            {
+                "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+                "threshold": "BLOCK_NONE"
+            },
+            {
+                "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+                "threshold": "BLOCK_NONE"
+            }
+        ]
+        
         response = gemini_model.generate_content(
             prompt_input,
-            generation_config=generation_config
+            generation_config=generation_config,
+            safety_settings=safety_settings
         )
         
+        # Check if response was blocked
+        if not response.text:
+            return f"⚠️ Response blocked by safety filters. Prompt feedback: {response.prompt_feedback}"
+        
         llm_response_text = response.text.strip()
         
         # Post-processing: Remove meta-commentary
@@ -265,8 +302,18 @@ def generate_feedback_from_prompt(prompt_input: str) -> str:
         return llm_response_text
         
     except Exception as e:
-        print(f"❌ Error during Gemini interaction: {e}")
-        return f"Error generating feedback: {str(e)}"
+        error_msg = str(e)
+        print(f"❌ Error during Gemini interaction: {error_msg}")
+        
+        # Provide helpful error messages
+        if "404" in error_msg or "not found" in error_msg:
+            return f"⚠️ Model not available. Please check your API key and model access. Error: {error_msg}"
+        elif "API key" in error_msg:
+            return "⚠️ Invalid API key. Please check your GEMINI_API_KEY in Spaces secrets."
+        elif "quota" in error_msg.lower():
+            return "⚠️ API quota exceeded. Please try again later or upgrade your API plan."
+        else:
+            return f"❌ Error generating feedback: {error_msg}"
 
 # ============================================
 # 9. GRADIO INTERFACE