Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - Flask Backend with Reliable Link Shortening
+# app.py - Complete AI Chat with Memory + Auto Link Shortening + Audio
 from flask import Flask, request, jsonify, send_from_directory, make_response
 import google.generativeai as genai
 from dotenv import load_dotenv
@@ -29,14 +29,17 @@ 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 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]
+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]
 """
 
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 model = genai.GenerativeModel('gemini-2.5-flash', system_instruction=system_instruction)
 
-# In-memory session storage
+# In-memory session storage (use Redis in production)
 chat_sessions = {}
 
 def convert_markdown_to_html(text):
@@ -62,37 +65,24 @@ def process_response(full_response):
 
 def generate_audio(text):
     """Generate audio file from text"""
-    clean_text = re.sub(r'[^\w\s.,!?\-]', '', text)
-    if not clean_text.strip():
-        clean_text = "Hello, this is your AI assistant."
+    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
+
+def shorten_url_with_tinyurl(url):
+    """Shorten URL using TinyURL public API"""
     try:
-        tts = gTTS(text=clean_text, lang='en', slow=False)
-        tts.save(filepath)
-        return filename
+        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
     except Exception as e:
-        print("TTS Error:", str(e))
+        print("TinyURL error:", 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:
@@ -108,23 +98,20 @@ def chat():
             chat_sessions[session_id] = model.start_chat(history=[])
         chat_session = chat_sessions[session_id]
 
-        # Always send to AI
+        # Send message to AI
         response = chat_session.send_message(user_message)
         full_text = response.text
 
-        # Force shortening if URL detected
+        # Auto-detect any URL and shorten it
         url_match = re.search(r'https?://[^\s<>"{}|\\^`\[\]]+', user_message)
         if url_match:
             original_url = url_match.group(0)
-            short_url = shorten_url_reliable(original_url)
-            if short_url:
+            short_url = shorten_url_with_tinyurl(original_url)
+            if short_url and "[SHORTEN]" not in full_text:
                 short_tag = f"[SHORTEN]{original_url}|{short_url}[/SHORTEN]"
-                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."
+                full_text += f"\n\n{short_tag}"
 
-        # Process final response
+        # Process response
         visible_text, audio_content = process_response(full_text)
         html_response = convert_markdown_to_html(visible_text)
 
@@ -133,24 +120,24 @@ def chat():
             "has_audio": False
         }
 
-        # Generate audio if available
+        # Generate audio if needed
         if audio_content:
             audio_filename = generate_audio(audio_content)
-            if audio_filename:
-                result["audio_filename"] = audio_filename
-                result["has_audio"] = True
+            result["audio_filename"] = audio_filename
+            result["has_audio"] = True
 
-        # Return with session cookie
+        # Return response + set 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("Server Error:", str(e))
-        return jsonify({"error": "Failed to get response. Try again."}), 500
+        print("Error:", str(e))
+        return jsonify({"error": "Something went wrong."}), 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')
@@ -161,7 +148,7 @@ def download_audio(filename):
     try:
         return send_from_directory(AUDIO_FOLDER, filename, as_attachment=True)
     except FileNotFoundError:
-        return jsonify({"error": "Audio file not found"}), 404
+        return jsonify({"error": "File not found"}), 404
 
 @app.route('/')
 def serve_index():