Update app.py

app.py

@@ -1,37 +1,50 @@
 import json
+import os
 from flask import Flask, request, jsonify
 from transformers import pipeline
 
-# Load the model (use a public model)
-llm = pipeline("text-generation", model="mistralai/Mistral-7B-v0.1")
+# Load a public model that doesn't require authentication
+llm = pipeline("text-generation", model="tiiuae/falcon-7b-instruct")
 
-# Load the schemes data from JSON
-with open("schemes.json", "r", encoding="utf-8") as f:
-    schemes_data = json.load(f)
+# Load the schemes data from JSON if available
+SCHEMES_FILE = "schemes.json"
+
+if os.path.exists(SCHEMES_FILE):
+    with open(SCHEMES_FILE, "r", encoding="utf-8") as f:
+        schemes_data = json.load(f)
+else:
+    schemes_data = {"schemes": []}  # Default to empty if file is missing
 
 app = Flask(__name__)
 
-# Function to search for relevant information
+# Function to search for relevant scheme information
 def find_scheme_info(query):
-    for scheme in schemes_data["schemes"]:
+    for scheme in schemes_data.get("schemes", []):
         if query.lower() in scheme["name"].lower() or query.lower() in scheme["description"].lower():
             return scheme["description"]
-    return "Sorry, I couldn't find information on that scheme."
+    return None  # Return None instead of a default message
 
-# Function to generate a chatbot response
+# Function to generate chatbot response
 def chatbot_response(query):
     scheme_info = find_scheme_info(query)
-    if scheme_info != "Sorry, I couldn't find information on that scheme.":
-        return scheme_info
-    else:
+    
+    if scheme_info:
+        return scheme_info  # Return scheme info if found
+    
+    try:
         response = llm(query, max_length=200, do_sample=True)
         return response[0]['generated_text']
+    except Exception as e:
+        return f"Error generating response: {str(e)}"
 
+# API route for chatbot
 @app.route("/chat", methods=["POST"])
 def chat():
-    user_input = request.json.get("query", "")
+    user_input = request.json.get("query", "").strip()
+    
     if not user_input:
         return jsonify({"error": "No query provided"}), 400
+    
     response = chatbot_response(user_input)
     return jsonify({"response": response})