Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py - Flask Backend with CORRECT spoo.me URL Shortening
 from flask import Flask, request, jsonify, send_from_directory, make_response
 import google.generativeai as genai
 from dotenv import load_dotenv
@@ -13,14 +12,14 @@ import uuid
 # Load environment variables
 load_dotenv()
 
-# Configure paths
+# Directories
 AUDIO_FOLDER = os.path.join('static', 'audio')
 os.makedirs(AUDIO_FOLDER, exist_ok=True)
 
 app = Flask(__name__, static_folder='static')
 CORS(app)
 
-# AI Configuration
+# Gemini AI Configuration
 system_instruction = """
 You are a helpful AI assistant named Athspi. When responding:
 1. Never mention "audio" or technical terms
@@ -73,43 +72,37 @@ def generate_audio(text):
     return filename
 
 def shorten_url_with_spoo_me(url):
-    """Shorten URL using spoo.me API - FIXED WITH STRIP & CORRECT JSON KEY"""
+    """Shorten URL using spoo.me API and clean output"""
     try:
-        # Clean the input URL
         clean_url = url.strip()
         if not clean_url.startswith(('http://', 'https://')):
             clean_url = 'https://' + clean_url
 
-        payload = {
-            "url": clean_url
-        }
+        payload = {"url": clean_url}
         headers = {
             "Accept": "application/json",
             "Content-Type": "application/x-www-form-urlencoded"
         }
 
-        response = requests.post(
-            "https://spoo.me/",
-            data=payload,
-            headers=headers,
-            timeout=10
-        )
+        response = requests.post("https://spoo.me/", data=payload, headers=headers, timeout=10)
 
         if response.status_code == 200:
             try:
-                result = response.json()  # This is now correct
-                short_url = result.get("short_url")  # Use "short_url", not "url"
+                result = response.json()
+                original_url = result.get("original_url", "").strip()
+                short_url = result.get("short_url", "").strip()
                 if short_url:
-                    return short_url.strip()  # CRITICAL: strip whitespace
+                    print(f"Spoo.me returned: {original_url} → {short_url}")
+                    return short_url
             except Exception as json_error:
-                print("JSON parse error:", str(json_error))
+                print("JSON parse error:", json_error)
                 return None
 
-        print(f"spoo.me error: {response.status_code} - {response.text}")
+        print(f"Spoo.me error: {response.status_code} - {response.text}")
         return None
 
     except Exception as e:
-        print("Request failed:", str(e))
+        print("Request failed:", e)
         return None
 
 @app.route('/chat', methods=['POST'])
@@ -127,20 +120,19 @@ def chat():
             chat_sessions[session_id] = model.start_chat(history=[])
         chat_session = chat_sessions[session_id]
 
-        # Send message to AI
+        # AI response
         response = chat_session.send_message(user_message)
         full_text = response.text
 
-        # Auto-detect any URL and shorten it
+        # Detect and shorten first URL
         url_match = re.search(r'https?://[^\s<>"{}|\\^`\[\]]+', user_message)
         if url_match:
-            original_url = url_match.group(0).strip()  # Strip detected URL
+            original_url = url_match.group(0).strip()
             short_url = shorten_url_with_spoo_me(original_url)
             if short_url and "[SHORTEN]" not in full_text:
                 short_tag = f"[SHORTEN]{original_url}|{short_url}[/SHORTEN]"
                 full_text += f"\n\n{short_tag}"
 
-        # Process final response
         visible_text, audio_content = process_response(full_text)
         html_response = convert_markdown_to_html(visible_text)
 
@@ -149,13 +141,12 @@ 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
 
-        # Return response + set session cookie
+        # Set session ID in cookie
         resp = make_response(jsonify(result))
         resp.set_cookie('session_id', session_id, max_age=3600, httponly=True, samesite='Lax')
         return resp
@@ -166,7 +157,6 @@ def chat():
 
 @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')
@@ -188,4 +178,4 @@ def serve_static(path):
     return send_from_directory('static', path)
 
 if __name__ == '__main__':
-    app.run(host="0.0.0.0", port=7860)
+    app.run(host="0.0.0.0", port=7860)