update remote

app.py

@@ -1,20 +1,20 @@
 from flask import Flask, render_template, request, jsonify
 import os
-from google.generativeai import generative_models as gen_models
+import google.generativeai as genai
 from dotenv import load_dotenv
+import markdown  # Import the markdown library
 
 load_dotenv()
 app = Flask(__name__)
 
 # Configure Gemini API key
-os.environ["GEMINI_API_KEY"] = os.getenv("GEMINI_API_KEY")
+genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
-model = gen_models.GenerativeModel("gemini-pro")  # Or "gemini-1.5-flash" if available
+model = genai.GenerativeModel("gemini-2.0-flash")  # Or "gemini-1.5-flash" if available
 
 chat_history = []
 
 @app.route("/", methods=["GET", "POST"])
-
 def index():
     response_text = None
     if request.method == "POST":
@@ -22,18 +22,21 @@ def index():
         if prompt:
             try:
                 chat_history.append({"role": "user", "parts": prompt})
-                response = model.generate_text(
-                    prompt=prompt,
-                    temperature=0.7,  # Adjust as needed
-                    max_output_tokens=800  # Adjust as needed
+                response = model.generate_content(
+                    prompt,
+                    generation_config=genai.GenerationConfig(
+                        temperature=0.7,  # Adjust as needed
+                        max_output_tokens=800  # Adjust as needed
+                    )
                 )
-                response_text = response.result
-                chat_history.append({"role": "model", "parts": response_text})
+                response_text = response.text
+                # Convert markdown to HTML
+                response_html = markdown.markdown(response_text)
+                chat_history.append({"role": "model", "parts": response_html}) #store html
             except Exception as e:
                 response_text = f"Error: {e}"
 
     return render_template("index.html", response_text=response_text, chat_history=chat_history)
 
-
 if __name__ == "__main__":
     app.run(debug=False, host='0.0.0.0', port=int(os.environ.get('PORT', 8080)))

requirements.txt

@@ -1,4 +1,5 @@
 Flask
 google-generativeai
 python-dotenv
-gunicorn  # For Hugging Face deployment
+gunicorn  # For Hugging Face deployment
+markdown

templates/index.html

@@ -9,7 +9,7 @@
     <div id="chat-area">
         {% for message in chat_history %}
             <div class="message {{ message.role }}">
-                <strong>{{ message.role.capitalize() }}:</strong> {{ message.parts }}
+                <strong>{{ message.role.capitalize() }}:</strong> {{ message.parts|safe }}
             </div>
         {% endfor %}
     </div>
@@ -20,11 +20,5 @@
         <input type="submit" value="Send">
     </form>
 
-    {% if response_text %}
-        <div class="message model">
-            <strong>Model:</strong> {{ response_text }}
-        </div>
-    {% endif %}
-
 </body>
-</html>
+</html>