Update app.py

app.py

@@ -8,6 +8,7 @@ import markdown2
 import re
 from gtts import gTTS
 import uuid
+from collections import deque
 
 # Load environment variables
 load_dotenv()
@@ -19,18 +20,18 @@ os.makedirs(AUDIO_FOLDER, exist_ok=True)
 app = Flask(__name__, static_folder='static')
 CORS(app)
 
+# Conversation history storage
+conversation_histories = {}
+MAX_HISTORY_LENGTH = 10  # Number of messages to remember
+
 # AI Configuration
 system_instruction = """
 You are a helpful AI assistant named Athspi. When responding:
-1. Never mention "audio" or technical terms
-2. For responses that would benefit from audio (like stories, explanations, or content meant to be heard), include the audio version between these markers:
+1. Maintain conversation context naturally
+2. For responses that would benefit from audio (like stories), include between:
    [AUDIO]content here[/AUDIO]
-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
-Example good response:
-Here's a story for you!
-[AUDIO]Once upon a time...[/AUDIO]
+3. Keep responses conversational and friendly
+4. Remember previous interactions in this conversation
 """
 
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
@@ -42,14 +43,12 @@ def convert_markdown_to_html(text):
     return html
 
 def process_response(full_response):
-    """Extract visible text and audio content"""
     audio_match = re.search(r'\[AUDIO\](.*?)\[/AUDIO\]', full_response, re.DOTALL)
     audio_content = audio_match.group(1).strip() if audio_match else None
     visible_text = re.sub(r'\[/?AUDIO\]', '', full_response).strip()
     return visible_text, audio_content
 
 def generate_audio(text):
-    """Generate audio file from text"""
     text = re.sub(r'[^\w\s.,!?\-]', '', text)
     filename = f"audio_{uuid.uuid4()}.mp3"
     filepath = os.path.join(AUDIO_FOLDER, filename)
@@ -57,30 +56,51 @@ def generate_audio(text):
     tts.save(filepath)
     return filename
 
+def get_conversation_history(session_id):
+    if session_id not in conversation_histories:
+        conversation_histories[session_id] = deque(maxlen=MAX_HISTORY_LENGTH)
+    return conversation_histories[session_id]
+
+@app.route('/start_session', methods=['POST'])
+def start_session():
+    session_id = str(uuid.uuid4())
+    conversation_histories[session_id] = deque(maxlen=MAX_HISTORY_LENGTH)
+    return jsonify({"session_id": session_id})
+
 @app.route('/chat', methods=['POST'])
 def chat():
     try:
         data = request.json
         user_message = data.get('message', '').strip()
+        session_id = data.get('session_id')
         
         if not user_message:
             return jsonify({"error": "Message required"}), 400
+        if not session_id:
+            return jsonify({"error": "Session ID required"}), 400
 
-        # Get AI response - the AI will automatically decide when to include audio
-        response = model.generate_content(user_message)
-        visible_text, audio_content = process_response(response.text)
+        history = get_conversation_history(session_id)
         
-        # Generate HTML response
-        html_response = convert_markdown_to_html(visible_text)
+        # Build conversation context
+        chat_session = model.start_chat(history=list(history))
+        
+        # Get AI response with context
+        response = chat_session.send_message(user_message)
+        
+        # Update history
+        history.extend([
+            {"role": "user", "parts": [user_message]},
+            {"role": "model", "parts": [response.text]}
+        ])
+        
+        visible_text, audio_content = process_response(response.text)
         
-        # Prepare result
         result = {
             "response_text": visible_text,
-            "response_html": html_response,
+            "response_html": convert_markdown_to_html(visible_text),
             "has_audio": False
         }
         
-        # Generate audio if AI included audio content
         if audio_content:
             audio_filename = generate_audio(audio_content)
             result["audio_filename"] = audio_filename