Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - Flask Backend with spoo.me URL Shortening + Chat Memory + Audio
+# app.py - Fixed URL Shortening with spoo.me
 from flask import Flask, request, jsonify, send_from_directory, make_response
 import google.generativeai as genai
 from dotenv import load_dotenv
@@ -32,14 +32,14 @@ You are a helpful AI assistant named Athspi. When responding:
 6. If a URL is present in the user's message, respond with:
    [SHORTEN]original_url|short_url[/SHORTEN]
    Example:
-   I've shortened your link for convenience:
+   I've shortened your link:
    [SHORTEN]https://example.com|https://spoo.me/abc123[/SHORTEN]
 """
 
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
-model = genai.GenerativeModel('gemini-2.5-flash', system_instruction=system_instruction)
+model = genai.GenerativeModel('gemini-2.5-flasho', system_instruction=system_instruction)
 
-# In-memory session storage (use Redis in production)
+# In-memory session storage
 chat_sessions = {}
 
 def convert_markdown_to_html(text):
@@ -65,33 +65,54 @@ def process_response(full_response):
 
 def generate_audio(text):
     """Generate audio file from text"""
-    text = re.sub(r'[^\w\s.,!?\-]', '', text)
+    clean_text = re.sub(r'[^\w\s.,!?\-]', '', text)
     filename = f"audio_{uuid.uuid4()}.mp3"
     filepath = os.path.join(AUDIO_FOLDER, filename)
-    tts = gTTS(text=text, lang='en', slow=False)
-    tts.save(filepath)
-    return filename
+    try:
+        tts = gTTS(text=clean_text, lang='en', slow=False)
+        tts.save(filepath)
+        return filename
+    except Exception as e:
+        print("TTS Error:", str(e))
+        return None
 
 def shorten_url_with_spoo_me(url):
-    """Shorten URL using spoo.me API"""
+    """Shorten URL using spoo.me API with correct headers"""
     try:
+        # Ensure URL has scheme
+        if not url.startswith("http"):
+            url = "https://" + url
+
         payload = {
-            "url": url,
-            "alias": ""  # Leave empty for auto-generated
+            "url": url
+            # alias can be added: "alias": "mycustom"
         }
         headers = {
             "Accept": "application/json",
             "Content-Type": "application/x-www-form-urlencoded"
         }
-        response = requests.post("https://spoo.me/", data=payload, headers=headers, timeout=5)
+
+        response = requests.post(
+            "https://spoo.me/",
+            data=payload,  # This sends as form-encoded
+            headers=headers,
+            timeout=10
+        )
+
+        print("spoo.me Status:", response.status_code)  # Debug
+        print("spoo.me Response:", response.text)      # Debug
+
         if response.status_code == 200:
             result = response.json()
-            return result.get("shorturl") or result.get("url")  # spoo.me returns either
-        else:
-            print("spoo.me error:", response.status_code, response.text)
-            return None
+            short_url = result.get("shorturl") or result.get("url")
+            if short_url and short_url.startswith("http"):
+                return short_url
+
+        print("Failed to shorten:", response.status_code, response.text)
+        return None
+
     except Exception as e:
-        print("Request failed:", str(e))
+        print("Request Exception:", str(e))
         return None
 
 @app.route('/chat', methods=['POST'])
@@ -109,18 +130,24 @@ def chat():
             chat_sessions[session_id] = model.start_chat(history=[])
         chat_session = chat_sessions[session_id]
 
-        # Send message to AI
+        # Send to AI first
         response = chat_session.send_message(user_message)
-        full_text = response.text
+        full_text = response.text.strip()
 
-        # Auto-detect any URL and shorten it using spoo.me
+        # Extract URL and shorten
         url_match = re.search(r'https?://[^\s<>"{}|\\^`\[\]]+', user_message)
         if url_match:
             original_url = url_match.group(0)
             short_url = shorten_url_with_spoo_me(original_url)
-            if short_url and "[SHORTEN]" not in full_text:
+
+            if short_url:
                 short_tag = f"[SHORTEN]{original_url}|{short_url}[/SHORTEN]"
-                full_text += f"\n\n{short_tag}"
+                if "[SHORTEN]" not in full_text:
+                    full_text += f"\n\n{short_tag}"
+            else:
+                # Optional: Inform user
+                # full_text += "\n\nI couldn't shorten the link right now."
+                pass  # Silent fallback
 
         # Process final response
         visible_text, audio_content = process_response(full_text)
@@ -131,24 +158,23 @@ def chat():
             "has_audio": False
         }
 
-        # Generate audio if needed
         if audio_content:
             audio_filename = generate_audio(audio_content)
-            result["audio_filename"] = audio_filename
-            result["has_audio"] = True
+            if audio_filename:
+                result["audio_filename"] = audio_filename
+                result["has_audio"] = True
 
-        # Return response + set session cookie
+        # Return response with session cookie
         resp = make_response(jsonify(result))
         resp.set_cookie('session_id', session_id, max_age=3600, httponly=True, samesite='Lax')
         return resp
 
     except Exception as e:
-        print("Error:", str(e))
-        return jsonify({"error": "Something went wrong."}), 500
+        print("Server Error:", str(e))
+        return jsonify({"error": "Something went wrong. Please try again."}), 500
 
 @app.route('/new-chat', methods=['POST'])
 def new_chat():
-    """Reset chat memory"""
     session_id = str(uuid.uuid4())
     resp = make_response(jsonify({"status": "new chat started"}))
     resp.set_cookie('session_id', session_id, max_age=3600, httponly=True, samesite='Lax')
@@ -159,7 +185,7 @@ def download_audio(filename):
     try:
         return send_from_directory(AUDIO_FOLDER, filename, as_attachment=True)
     except FileNotFoundError:
-        return jsonify({"error": "File not found"}), 404
+        return jsonify({"error": "Audio file not found"}), 404
 
 @app.route('/')
 def serve_index():