Update app.py

app.py

@@ -1,23 +1,36 @@
-from fastapi import FastAPI, HTTPException
+# app.py
+from flask import Flask, request, jsonify
 import openai
 
-# Initialize the FastAPI app
-app = FastAPI()
+app = Flask(__name__)
 
-# Set your OpenAI API key directly here
+# API Key yahan enter karo
 openai.api_key = "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"
 
-@app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
+@app.route("/", methods=["GET"])
+def home():
+    return "<h1>ChatGPT Query Service</h1><p>Enter your query at /query endpoint.</p>"
 
-@app.post("/chat")
-async def chat_with_gpt(prompt: str):
+@app.route("/query", methods=["POST"])
+def query():
+    data = request.get_json()
+    query_text = data.get("query", "")
+    
+    if not query_text:
+        return jsonify({"error": "Query text is required"}), 400
+    
     try:
-        response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",
-            messages=[{"role": "user", "content": prompt}]
+        response = openai.Completion.create(
+            engine="text-davinci-003",
+            prompt=query_text,
+            max_tokens=100
         )
-        return {"response": response.choices[0].message['content']}
+        
+        answer = response.choices[0].text.strip()
+        return jsonify({"query": query_text, "answer": answer})
+    
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        return jsonify({"error": str(e)}), 500
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=8000)