Update app.py

app.py

@@ -1,32 +1,25 @@
 import os
-import sys
-import asyncio
+import openai
 import logging
-import openai  # Correct OpenAI import
-from transformers import AutoTokenizer
-from typing import Dict, Any, List
-import aiohttp
-from cryptography.fernet import Fernet
 import gradio as gr
-import json
-import torch
-import psutil
-import random
+import asyncio
+from typing import Dict, Any
+from cryptography.fernet import Fernet
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
 
 class EnvironmentManager:
-    """Handles loading and validation of environment variables."""
+    """Handles environment variable validation."""
     @staticmethod
     def load_env_variables() -> Dict[str, str]:
-        required_vars = [
-            "OPENAI_API_KEY", "ENCRYPTION_KEY"
-        ]
-
+        required_vars = ["OPENAI_API_KEY", "ENCRYPTION_KEY"]
         env_vars = {var: os.getenv(var) for var in required_vars}
+        
         missing_vars = [var for var, value in env_vars.items() if not value]
-
         if missing_vars:
-            raise ValueError(f"Missing required environment variables: {', '.join(missing_vars)}")
-
+            raise ValueError(f"Missing environment variables: {', '.join(missing_vars)}")
+        
         return env_vars
 
 class EncryptionManager:
@@ -41,7 +34,7 @@ class EncryptionManager:
         return self.cipher.decrypt(encrypted_data.encode()).decode()
 
 class AICore:
-    """Main AI Core system integrating OpenAI chat functionality."""
+    """AI Core system integrating OpenAI's GPT API."""
     def __init__(self, env_vars: Dict[str, str]):
         self.env_vars = env_vars
         self.encryption_manager = EncryptionManager(env_vars["ENCRYPTION_KEY"])
@@ -51,8 +44,8 @@ class AICore:
         try:
             encrypted_query = self.encryption_manager.encrypt(query)
 
-            chat_completion = await openai.ChatCompletion.acreate(
-                model="gpt-4-turbo",  # Ensure you use a valid model name
+            response = await openai.ChatCompletion.acreate(
+                model="ft:gpt-4o-2024-08-06:raiffs-bits:codettev7",  # Ensure this model is supported
                 messages=[
                     {"role": "system", "content": "You are a helpful AI assistant."},
                     {"role": "user", "content": query}
@@ -60,6 +53,41 @@ class AICore:
                 api_key=self.openai_api_key
             )
 
-            model_response = chat_completion['choices'][0]['message']['content']
+            model_response = response["choices"][0]["message"]["content"]
+
+            return {
+                "encrypted_query": encrypted_query,
+                "model_response": model_response
+            }
+        except Exception as e:
+            logging.error(f"Error generating response: {e}")
+            return {"error": "Failed to generate response"}
+
+# Hugging Face Gradio App
+def main():
+    try:
+        env_vars = EnvironmentManager.load_env_variables()
+        ai_core = AICore(env_vars)
+
+        async def async_respond(message: str) -> str:
+            response_data = await ai_core.generate_response(message)
+            return response_data.get("model_response", "Error: Response not available")
+
+        def respond(message: str) -> str:
+            return asyncio.run(async_respond(message))
+
+        interface = gr.Interface(
+            fn=respond,
+            inputs="text",
+            outputs="text",
+            title="AI Chatbot - Hugging Face Space"
+        )
+
+        # Use `server_name="0.0.0.0"` to work properly on Hugging Face Spaces
+        interface.launch(server_name="0.0.0.0", server_port=7860)
+
+    except Exception as e:
+        logging.error(f"Application failed to start: {e}")
 
-            return
+if __name__ == "__main__":
+    main()