Update handler.py

handler.py

@@ -3,21 +3,6 @@ import openai
 from environs import Env
 from typing import List, Dict, Any
 
-
-import requests
-
-def download_env_file(url: str, local_path: str):
-    response = requests.get(url)
-    response.raise_for_status()  # Ensure we notice bad responses
-    with open(local_path, 'wb') as f:
-        f.write(response.content)
-
-# Download the .env file
-env_file_url = "https://www.dropbox.com/scl/fi/21ldek2cdsak2v3mhyy5x/openai.env?rlkey=nxdkd8l8esdy8npa3vfgvqkhp&st=s2f2zzwl&dl=1"  # Adjusted URL for direct download
-local_env_path = "openai.env"
-download_env_file(env_file_url, local_env_path)
-
-
 # Load environment variables
 env = Env()
 env.read_env("openai.env")
@@ -28,7 +13,7 @@ MODEL = env.str("MODEL", "gpt-3.5-turbo")
 AI_RESPONSE_TIMEOUT = env.int("AI_RESPONSE_TIMEOUT", 20)
 
 class EndpointHandler:
-    def __init__(self, model_dir):
+    def __init__(self):
         pass
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
@@ -40,8 +25,6 @@ class EndpointHandler:
         except Exception as e:
             return {"error": str(e)}
 
-    # Rest of the code remains the same
-
     def process_json_input(self, json_data):
         if "FromUserKavasQuestions" in json_data and "Chatmood" in json_data:
             prompt = self.create_conversation_starter_prompt(
@@ -60,6 +43,16 @@ class EndpointHandler:
         else:
             raise ValueError("Invalid JSON structure.")
 
+    def process_hf_input(self, hf_data):
+        print("Received HF Data:", hf_data)  # Debugging line
+        if "inputs" in hf_data:
+            actual_data = hf_data["inputs"]
+            print("Processing actual data:", actual_data)  # Debugging line
+            return self.process_json_input(actual_data)
+        else:
+            return {"error": "Invalid Hugging Face JSON structure."}
+
+
     def create_conversation_starter_prompt(self, user_questions, chatmood):
         formatted_info = " ".join([f"{qa['Question']} - {qa['Answer']}" for qa in user_questions if qa['Answer']])
         prompt = (f"Based on user profile info and a {chatmood} mood, "