Update app.py

app.py

@@ -1,65 +1,18 @@
-import google.generativeai as genai
-import os
-from config import (
-    SYSTEM_MESSAGE,
-    MODEL_NAME,
-    DEFAULT_MAX_NEW_TOKENS,
-    DEFAULT_TEMPERATURE,
-    DEFAULT_TOP_P,
-    DEFAULT_TOP_K,
-    DEFAULT_STOP_SEQUENCES,
-)
+import asyncio
+import nest_asyncio
+from gradio_interface import create_gradio_interface
+from discord_bot import run_discord_bot
 
-# Load Google API key from environment
-GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
-if not GOOGLE_API_KEY:
-    raise ValueError("GOOGLE_API_KEY is not set. Please provide your API key.")
+# Apply nest_asyncio to allow nested event loops
+nest_asyncio.apply()
 
-genai.configure(api_key=GOOGLE_API_KEY)
+if __name__ == "__main__":
+    # Create the event loop
+    loop = asyncio.get_event_loop()
 
-# Query AI model
-async def query_ai_model(
-    message,
-    history=None,
-    system_message=SYSTEM_MESSAGE,
-    max_new_tokens=DEFAULT_MAX_NEW_TOKENS,
-    temperature=DEFAULT_TEMPERATURE,
-    top_p=DEFAULT_TOP_P,
-    top_k=DEFAULT_TOP_K,
-    stop_sequences=DEFAULT_STOP_SEQUENCES,
-):
-    try:
-        # Build the conversation history in the required format
-        messages = []
+    # Start the Discord bot in the event loop
+    loop.create_task(run_discord_bot())
 
-        if system_message:
-            messages.append({'role': 'system', 'content': system_message})
-
-        if history:
-            for user_msg, bot_reply in history:
-                if user_msg:
-                    messages.append({'role': 'user', 'content': user_msg})
-                if bot_reply:
-                    messages.append({'role': 'assistant', 'content': bot_reply})
-
-        # Append the new user message
-        messages.append({'role': 'user', 'content': message})
-
-        # Set parameters
-        parameters = {
-            'temperature': temperature,
-            'top_p': top_p,
-            'top_k': top_k,
-            'max_output_tokens': int(max_new_tokens),
-            'stop_sequences': stop_sequences,
-        }
-
-        # Generate response
-        response = genai.generate_chat(
-            model=MODEL_NAME,
-            messages=messages,
-            **parameters
-        )
-        return response.candidates[0]['content']
-    except Exception as e:
-        return f"An error occurred: {str(e)}"
+    # Create and launch the Gradio interface
+    demo = create_gradio_interface()
+    demo.launch(share=True)