Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - Flask Backend with AI URL Shortening System
+# app.py - Flask Backend with URL Shortening System
 from flask import Flask, request, jsonify, send_from_directory, make_response
 import google.generativeai as genai
 from dotenv import load_dotenv
@@ -35,10 +35,11 @@ You are a helpful AI assistant named Athspi. When responding:
    [SHORTEN]https://example.com|https://spoo.me/abc123[/SHORTEN]
 7. DO NOT add any extra text inside the SHORTEN tags
 8. ONLY use this format when you have a shortened URL to provide
+9. If you cannot shorten the URL, do not include the SHORTEN tags
 """
 
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
-model = genai.GenerativeModel('gemini-2.5-flash', system_instruction=system_instruction)
+model = genai.GenerativeModel('gemini-1.5-pro', system_instruction=system_instruction)
 
 # In-memory session storage
 chat_sessions = {}
@@ -107,18 +108,19 @@ def shorten_url_with_spoo_me(url):
 
         if response.status_code == 200:
             try:
-                # Parse the JSON response
+                # Parse the JSON response - THIS IS THE CORRECT WAY
                 result = response.json()
                 # Extract short_url and strip whitespace
                 short_url = result.get("short_url", "").strip()
                 if short_url and short_url.startswith('http'):
                     return short_url
-            except Exception as json_error:
-                print("JSON parse error:", str(json_error))
-                # Fallback for plain text response
+            except Exception:
+                # If JSON parsing fails, try plain text response
                 text_response = response.text.strip()
                 if text_response.startswith('http'):
                     return text_response
+                elif len(text_response) > 3:
+                    return f"https://spoo.me/{text_response}"
 
         print(f"spoo.me error: {response.status_code} - {response.text}")
         return None
@@ -144,7 +146,6 @@ def chat():
 
         # Check if user message contains a URL
         url_match = re.search(r'https?://[^\s<>"{}|\\^`\[\]]+', user_message)
-        ai_response = None
         full_text = ""
         
         if url_match:
@@ -157,10 +158,10 @@ def chat():
                 
                 # Send message to AI with instruction to include short link
                 ai_response = chat_session.send_message(
-                    f"User sent this message: '{user_message}'\n\n"
-                    f"I found a URL: {original_url}\n"
-                    f"Shortened URL: {short_url}\n\n"
-                    f"Respond naturally and include this exact format:\n{short_tag}"
+                    f"{user_message}\n\n"
+                    f"I detected a URL in your message: {original_url}\n"
+                    f"Shortened version: {short_url}\n\n"
+                    f"Please respond naturally and include this exact format:\n{short_tag}"
                 )
                 full_text = ai_response.text
             else: