ai

main.py

@@ -2,12 +2,33 @@ import os
 from fastapi import FastAPI, Request, HTTPException, Query
 from fastapi.responses import PlainTextResponse, JSONResponse
 import httpx
+from openai import OpenAI
+from pydantic import BaseModel
+from typing import Optional
 
 app = FastAPI()
 
 VERIFY_TOKEN = os.getenv("VERIFY_TOKEN", "")
 PHONE_NUMBER_ID = os.getenv("PHONE_NUMBER_ID", "")
 WHATSAPP_API_TOKEN = os.getenv("WHATSAPP_API_TOKEN", "")
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
+
+# Initialize OpenAI client
+openai_client = None
+if OPENAI_API_KEY:
+    openai_client = OpenAI(api_key=OPENAI_API_KEY)
+
+# Pydantic model for chat requests
+class ChatRequest(BaseModel):
+    message: str
+    model: Optional[str] = "gpt-3.5-turbo"  # Using the cheapest OpenAI model
+    max_tokens: Optional[int] = 150
+
+# Pydantic model for chat responses
+class ChatResponse(BaseModel):
+    response: str
+    model: str
+    tokens_used: Optional[int] = None
 
 @app.get("/webhook")
 def verify_webhook(
@@ -35,8 +56,14 @@ async def receive_message(req: Request):
         msg = messages[0]
 
         if msg.get("type") == "text":
-            reply_text = "Hello 👋 from PropAgent Git!"
-            await send_whatsapp_message(wa_id, reply_text)
+            # Get the user's message
+            user_message = msg.get("text", {}).get("body", "")
+            
+            # Generate AI response using OpenAI
+            ai_response = await generate_ai_response(user_message)
+            
+            # Send the AI response back to WhatsApp
+            await send_whatsapp_message(wa_id, ai_response)
             return JSONResponse({"status": "replied"})
 
     except Exception as e:
@@ -45,6 +72,58 @@ async def receive_message(req: Request):
 
     return JSONResponse({"status": "ignored"})
 
+async def generate_ai_response(user_message: str) -> str:
+    """Generate AI response using OpenAI API"""
+    if not openai_client:
+        return "Sorry, AI chat is not available. Please check your OpenAI API key configuration."
+    
+    try:
+        response = openai_client.chat.completions.create(
+            model="gpt-3.5-turbo",
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant. Keep your responses concise and friendly."},
+                {"role": "user", "content": user_message}
+            ],
+            max_tokens=150,
+            temperature=0.7
+        )
+        
+        ai_response = response.choices[0].message.content
+        return ai_response if ai_response else "I'm sorry, I couldn't generate a response."
+        
+    except Exception as e:
+        print(f"Error generating AI response: {e}")
+        return "Sorry, I'm having trouble processing your request right now."
+
+@app.post("/chat", response_model=ChatResponse)
+async def chat_with_ai(request: ChatRequest):
+    """Basic chat endpoint for direct API calls"""
+    if not openai_client:
+        raise HTTPException(status_code=500, detail="OpenAI API key not configured")
+    
+    try:
+        response = openai_client.chat.completions.create(
+            model=request.model,
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant. Keep your responses concise and friendly."},
+                {"role": "user", "content": request.message}
+            ],
+            max_tokens=request.max_tokens,
+            temperature=0.7
+        )
+        
+        ai_response = response.choices[0].message.content
+        usage = response.usage
+        
+        return ChatResponse(
+            response=ai_response,
+            model=request.model,
+            tokens_used=usage.total_tokens if usage else None
+        )
+        
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error generating response: {str(e)}")
+
 async def send_whatsapp_message(wa_id: str, message: str):
     url = f"https://graph.facebook.com/v18.0/{PHONE_NUMBER_ID}/messages"
     headers = {
@@ -65,4 +144,14 @@ async def send_whatsapp_message(wa_id: str, message: str):
 
 @app.get("/ping")
 def ping():
-    return {"pong": True}
+    return {"pong": True}
+
+@app.get("/health")
+def health_check():
+    """Health check endpoint to verify API connectivity"""
+    status = {
+        "api": "healthy",
+        "whatsapp": "configured" if WHATSAPP_API_TOKEN else "not configured",
+        "openai": "configured" if OPENAI_API_KEY else "not configured"
+    }
+    return status

requirements.txt

@@ -1,4 +1,5 @@
 fastapi
 uvicorn
 python-dotenv
-httpx 
+httpx
+openai