Update app.py

app.py

@@ -1,5 +1,4 @@
 import os
-import json
 import random
 import string
 import subprocess
@@ -11,22 +10,17 @@ import markdown
 import threading
 from queue import Queue
 import time
-import uuid
 
 app = Flask(__name__)
 
 # Define directories
 file_folder = os.path.dirname(os.path.abspath(__file__))
 temp_audio_folder = os.path.join(file_folder, 'temp_audio')
-chats_folder = os.path.join(file_folder, 'chats')
-sessions_folder = os.path.join(file_folder, 'sessions')
 model_folder = None
 piper_binary_path = os.path.join(file_folder, 'piper')
 
 # Create necessary directories
 os.makedirs(temp_audio_folder, exist_ok=True)
-os.makedirs(chats_folder, exist_ok=True)
-os.makedirs(sessions_folder, exist_ok=True)
 
 # Check default user folder
 default_user_folder = "./"
@@ -57,27 +51,6 @@ def get_ollama_models(base_host=DEFAULT_BASE_HOST):
     except:
         return []
 
-def load_chat_history(session_id):
-    try:
-        session_file = os.path.join(sessions_folder, f'{session_id}.json')
-        with open(session_file, 'r', encoding='utf-8') as f:
-            return json.load(f).get('messages', [])
-    except:
-        return []
-
-def save_chat(session_id, messages):
-    if not messages:
-        return None
-
-    session_file = os.path.join(sessions_folder, f'{session_id}.json')
-    with open(session_file, 'w', encoding='utf-8') as f:
-        json.dump({
-            'timestamp': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
-            'messages': messages
-        }, f, ensure_ascii=False, indent=2)
-
-    return session_id
-
 def remove_markdown(text):
     html_content = markdown.markdown(text)
     soup = BeautifulSoup(html_content, 'html.parser')
@@ -119,68 +92,22 @@ def convert_to_speech(text, model_name, remove_md=False):
 
     return None
 
-def set_default_models():
-    tts_models = get_available_models()
-    ollama_models = get_ollama_models()
-
-    default_tts_model = "RecomendacionesConMiau" if "RecomendacionesConMiau" in tts_models else None
-    default_ollama_model = "llama3.2:1b" if "llama3.2:1b" in ollama_models else None
-
-    return default_tts_model, default_ollama_model
-
 @app.route('/')
 def index():
-    session_id = str(uuid.uuid4())
     tts_models = get_available_models()
-    chat_files = sorted([f for f in os.listdir(chats_folder) if f.endswith('.json')], reverse=True)
-    default_tts_model, default_ollama_model = set_default_models()
-    return render_template('index.html', tts_models=tts_models, chat_files=chat_files, default_tts_model=default_tts_model, default_ollama_model=default_ollama_model, session_id=session_id)
+    return render_template('index.html', tts_models=tts_models)
 
 @app.route('/api/list_ollama_models')
 def list_ollama_models():
     base_host = request.args.get('base_host', DEFAULT_BASE_HOST)
     return jsonify(models=get_ollama_models(base_host))
 
-@app.route('/api/load_chat/<session_id>')
-def load_chat(session_id):
-    messages = load_chat_history(session_id)
-    return jsonify(messages=messages)
-
-@app.route('/api/update_message', methods=['POST'])
-def update_message():
-    data = request.json
-    session_id = data.get('session_id')
-    message_index = data.get('message_index')
-    new_content = data.get('content')
-    is_user = data.get('is_user', False)
-
-    if not session_id or message_index is None or not new_content:
-        return jsonify(error="Missing required parameters"), 400
-
-    messages = load_chat_history(session_id)
-    if message_index >= len(messages):
-        return jsonify(error="Invalid message index"), 400
-
-    # Update the message content
-    messages[message_index]['content'] = new_content
-
-    # If it's a user message, regenerate all subsequent responses
-    if is_user:
-        # Keep messages up to and including the edited message
-        messages = messages[:message_index + 1]
-
-    # Save the updated chat
-    save_chat(session_id, messages)
-
-    return jsonify(success=True, messages=messages)
-
 @app.route('/api/chat', methods=['POST'])
 def chat():
     data = request.json
     base_host = data.get('base_host', DEFAULT_BASE_HOST)
     model = data.get('model')
     messages = data.get('messages', [])
-    session_id = data.get('session_id')
 
     def generate():
         queue = Queue()
@@ -200,9 +127,6 @@ def chat():
                 complete_response = content
                 yield f"data: {json.dumps({'chunk': content})}\n\n"
             elif msg_type == "done":
-                # Save chat history
-                messages.append({"role": "assistant", "content": complete_response})
-                save_chat(session_id, messages)
                 yield f"data: {json.dumps({'done': complete_response})}\n\n"
                 break