Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - Complete AI Chat with Memory + Auto Link Shortening + Audio
+# app.py - Flask Backend with Reliable Link Shortening
 from flask import Flask, request, jsonify, send_from_directory, make_response
 import google.generativeai as genai
 from dotenv import load_dotenv
@@ -29,17 +29,14 @@ You are a helpful AI assistant named Athspi. When responding:
 3. Keep responses natural and friendly
 4. Decide automatically when to include audio based on the content type
 5. For stories, always include audio version
-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:
-   [SHORTEN]https://example.com|https://tinyurl.com/abc123[/SHORTEN]
+6. If a URL is in the user's message, summarize it and say you've shortened it.
+   Example: "Here's your shortened link:" followed by [SHORTEN]original|short[/SHORTEN]
 """
 
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 model = genai.GenerativeModel('gemini-2.5-flash', 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,24 +62,37 @@ 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)
+    if not clean_text.strip():
+        clean_text = "Hello, this is your AI assistant."
     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
-
-def shorten_url_with_tinyurl(url):
-    """Shorten URL using TinyURL public API"""
     try:
-        response = requests.get(f"http://tinyurl.com/api-create.php?url={url}")
-        if response.status_code == 200 and response.text.strip():
-            return response.text.strip()
-        return None
+        tts = gTTS(text=clean_text, lang='en', slow=False)
+        tts.save(filepath)
+        return filename
     except Exception as e:
-        print("TinyURL error:", e)
+        print("TTS Error:", str(e))
         return None
 
+def shorten_url_reliable(url):
+    """Try multiple services to shorten URL"""
+    services = [
+        f"https://tinyurl.com/api-create.php?url={url}",
+        f"https://is.gd/create.php?format=simple&url={url}"
+    ]
+    for service_url in services:
+        try:
+            response = requests.get(service_url, timeout=5)
+            if response.status_code == 200:
+                result = response.text.strip()
+                if result.startswith("http") and len(result) > 10:
+                    return result
+        except Exception as e:
+            print(f"Shortener failed: {service_url} | Error: {e}")
+            continue
+    return None
+
 @app.route('/chat', methods=['POST'])
 def chat():
     try:
@@ -98,20 +108,23 @@ def chat():
             chat_sessions[session_id] = model.start_chat(history=[])
         chat_session = chat_sessions[session_id]
 
-        # Send message to AI
+        # Always send to AI
         response = chat_session.send_message(user_message)
         full_text = response.text
 
-        # Auto-detect any URL and shorten it
+        # Force shortening if URL detected
         url_match = re.search(r'https?://[^\s<>"{}|\\^`\[\]]+', user_message)
         if url_match:
             original_url = url_match.group(0)
-            short_url = shorten_url_with_tinyurl(original_url)
-            if short_url and "[SHORTEN]" not in full_text:
+            short_url = shorten_url_reliable(original_url)
+            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\nHere's your shortened link:\n{short_tag}"
+            else:
+                full_text += "\n\nI found a link but couldn't shorten it right now."
 
-        # Process response
+        # Process final response
         visible_text, audio_content = process_response(full_text)
         html_response = convert_markdown_to_html(visible_text)
 
@@ -120,24 +133,24 @@ def chat():
             "has_audio": False
         }
 
-        # Generate audio if needed
+        # Generate audio if available
         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 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": "Failed to get response. 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')
@@ -148,7 +161,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():