Update app.py

app.py

@@ -224,8 +224,9 @@
 #     return {"message": "Welcome to the API"}
 
 import os
-import time
-import requests
+import datetime
+import json
+import logging
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.responses import HTMLResponse, JSONResponse
 from fastapi.staticfiles import StaticFiles
@@ -233,10 +234,8 @@ from pydantic import BaseModel
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.templating import Jinja2Templates
 from simple_salesforce import Salesforce, SalesforceLogin
-import uuid
-import datetime
-import json
-import logging
+from langchain_groq import ChatGroq
+from langchain_core.prompts import ChatPromptTemplate
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -252,7 +251,7 @@ app = FastAPI()
 # Allow CORS requests from any domain (not recommended for production)
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # Allows all origins
+    allow_origins=["*"],
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -263,16 +262,25 @@ app.mount("/static", StaticFiles(directory="static"), name="static")
 templates = Jinja2Templates(directory="static")
 
 # Validate environment variables
-required_env_vars = ["CHATGROQ_API_KEY", "username", "password", "security_token", "domain"]
+required_env_vars = ["GROQ_API_KEY", "username", "password", "security_token", "domain"]
 for var in required_env_vars:
     if not os.getenv(var):
         logger.error(f"Environment variable {var} is not set")
         raise ValueError(f"Environment variable {var} is not set")
 
-# Configure ChatGroq API (replace with actual endpoint and model)
-CHATGROQ_API_URL = os.getenv("CHATGROQ_API_URL", "https://api.example.com/v1/chat/completions")  # Update with correct endpoint
-CHATGROQ_API_KEY = os.getenv("CHATGROQ_API_KEY")
-CHATGROQ_MODEL = os.getenv("CHATGROQ_MODEL", "default-model")  # Update with correct model
+# Initialize Groq model
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
+GROQ_MODEL = "llama3-8b-8192"  # Default to a valid Groq model
+try:
+    llm = ChatGroq(
+        model_name=GROQ_MODEL,
+        api_key=GROQ_API_KEY,
+        temperature=0.1,
+        max_tokens=50
+    )
+except Exception as e:
+    logger.error(f"Failed to initialize Groq model: {e}")
+    raise HTTPException(status_code=500, detail="Failed to initialize Groq model")
 
 # Salesforce credentials
 username = os.getenv("username")
@@ -303,35 +311,24 @@ def handle_query(query):
             context_str += f"User asked: '{past_query}'\nBot answered: '{response}'\n"
 
     # Construct the prompt
-    prompt = f"""
-    You are the Clara Redfernstech chatbot. Provide accurate, professional answers in 10-15 words.
-    
-    Context:
-    {context_str}
-    
-    Question:
-    {query}
-    """
-
-    # Send request to ChatGroq API
-    headers = {
-        "Authorization": f"Bearer {CHATGROQ_API_KEY}",
-        "Content-Type": "application/json",
-    }
-    payload = {
-        "model": CHATGROQ_MODEL,  # Use configured model
-        "messages": [{"role": "user", "content": prompt}],
-        "max_tokens": 50,
-        "temperature": 0.1,
-    }
+    prompt_template = ChatPromptTemplate.from_messages([
+        ("system", """
+        You are the Clara Redfernstech chatbot. Provide accurate, professional answers in 10-15 words.
+        
+        Context:
+        {context}
+        
+        Question:
+        {query}
+        """),
+    ])
+    prompt = prompt_template.format(context=context_str, query=query)
 
     try:
-        response = requests.post(CHATGROQ_API_URL, headers=headers, json=payload, timeout=10)
-        response.raise_for_status()  # Raise exception for bad status codes
-        response_data = response.json()
-        response_text = response_data["choices"][0]["message"]["content"].strip()
-    except requests.exceptions.RequestException as e:
-        logger.error(f"Error querying ChatGroq API: {e}")
+        response = llm.invoke(prompt)
+        response_text = response.content.strip()
+    except Exception as e:
+        logger.error(f"Error querying Groq API: {e}")
         response_text = "Sorry, I couldn't find an answer."
 
     # Update chat history
@@ -388,7 +385,7 @@ async def receive_form_data(request: Request):
         return JSONResponse({"id": unique_id})
     except Exception as e:
         logger.error(f"Failed to create lead: {e}")
-        raise HTTPException(status_code=500, detail="Failed to create lead: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Failed to create lead: {str(e)}")
 
 @app.post("/chat/")
 async def chat(request: MessageRequest):