Adding Docker File

app.py

@@ -1,11 +1,14 @@
-import openai
+import requests
 import os
 from flask import Flask, request, jsonify, render_template
 
 app = Flask(__name__)
 
-# Set your OpenAI API key
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# Set your Hugging Face API key
+HF_API_KEY = os.getenv("HF_API_KEY")  # Store in environment variable
+
+API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct"
+HEADERS = {"Authorization": f"Bearer {HF_API_KEY}"}
 
 @app.route('/')
 def home():
@@ -17,19 +20,19 @@ def chat():
     
     if not user_message:
         return jsonify({"error": "Empty message received"})
-    
+
     try:
-        client = openai.OpenAI()  # New API usage
-        response = client.chat.completions.create(
-            model="gpt-3.5-turbo",
-            messages=[{"role": "user", "content": user_message}]
-        )
-        reply = response.choices[0].message.content
+        response = requests.post(API_URL, headers=HEADERS, json={"inputs": user_message})
+        data = response.json()
+        
+        if "error" in data:
+            return jsonify({"reply": f"Error: {data['error']}"})
+
+        reply = data[0]['generated_text'] if isinstance(data, list) else data.get('generated_text', "No response")
         return jsonify({"reply": reply})
 
-    except openai.OpenAIError as e:
-        print(f"OpenAI API Error: {e}")  # Print error for debugging
-        return jsonify({"reply": f"Error in response: {str(e)}"})  # Show error in UI
+    except Exception as e:
+        return jsonify({"reply": f"Error: {str(e)}"})
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)