Update app.py

app.py

@@ -6,8 +6,6 @@ import os
 from flask_cors import CORS
 import markdown2
 import re
-import tempfile
-from werkzeug.utils import secure_filename
 
 # Load environment variables
 load_dotenv()
@@ -16,29 +14,22 @@ load_dotenv()
 app = Flask(__name__, static_folder='static')
 CORS(app)  # Enable CORS for all routes
 
-# Configuration
-UPLOAD_FOLDER = 'uploads'
-os.makedirs(UPLOAD_FOLDER, exist_ok=True)
-ALLOWED_EXTENSIONS = {'pdf', 'txt', 'docx', 'pptx', 'xlsx', 'jpg', 'jpeg', 'png'}
-app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
-
 # Configure Gemini
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
-model = genai.GenerativeModel('gemini-2.5-flash')
-vision_model = genai.GenerativeModel('gemini-2.5-flash')
-
-def allowed_file(filename):
-    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+model = genai.GenerativeModel('gemini-pro')
 
 def convert_markdown_to_html(text):
     # Convert markdown to HTML
     html = markdown2.markdown(text)
     
-    # Add custom styling
+    # Add custom styling to code blocks
     html = html.replace('<code>', '<code class="code-block">')
+    
+    # Convert **bold** to <strong> for better visibility
     html = re.sub(r'\*\*(.*?)\*\*', r'<strong>\1</strong>', html)
+    
+    # Convert *italic* to <em>
     html = re.sub(r'\*(.*?)\*', r'<em>\1</em>', html)
-    html = html.replace('<a ', '<a target="_blank" rel="noopener noreferrer" ')
     
     return html
 
@@ -47,25 +38,12 @@ def chat():
     try:
         data = request.json
         user_message = data.get('message')
-        file_path = data.get('file_path')
         
-        if not user_message and not file_path:
-            return jsonify({"error": "No message or file provided"}), 400
+        if not user_message:
+            return jsonify({"error": "No message provided"}), 400
         
-        # Generate response based on input type
-        if file_path:
-            # Process file with vision model if it's an image
-            if file_path.lower().endswith(('.jpg', '.jpeg', '.png')):
-                response = vision_model.generate_content([user_message, genai.upload_file(file_path)])
-            else:
-                # Process text-based files
-                with open(file_path, 'r') as file:
-                    file_content = file.read()
-                prompt = f"{user_message}\n\nFile content:\n{file_content}"
-                response = model.generate_content(prompt)
-        else:
-            # Regular text chat
-            response = model.generate_content(user_message)
+        # Generate response using Gemini
+        response = model.generate_content(user_message)
         
         # Convert markdown to HTML with styling
         formatted_response = convert_markdown_to_html(response.text)
@@ -77,29 +55,6 @@ def chat():
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-@app.route('/upload', methods=['POST'])
-def upload_file():
-    try:
-        if 'file' not in request.files:
-            return jsonify({"error": "No file part"}), 400
-            
-        file = request.files['file']
-        if file.filename == '':
-            return jsonify({"error": "No selected file"}), 400
-            
-        if file and allowed_file(file.filename):
-            filename = secure_filename(file.filename)
-            file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
-            file.save(file_path)
-            return jsonify({
-                "message": "File uploaded successfully",
-                "file_path": file_path
-            })
-        else:
-            return jsonify({"error": "File type not allowed"}), 400
-    except Exception as e:
-        return jsonify({"error": str(e)}), 500
-
 @app.route('/')
 def serve_index():
     return send_from_directory('static', 'index.html')
@@ -109,4 +64,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=5000)