Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -5,13 +5,25 @@ from openai import OpenAI
|
|
5 |
from dotenv import load_dotenv
|
6 |
|
7 |
# Load API keys from .env file
|
8 |
-
load_dotenv()
|
9 |
-
API_KEY_LLAMA = os.getenv("OPENROUTER_API_KEY1") # Llama API Key
|
10 |
-
API_KEY_GEMMA = os.getenv("OPENROUTER_API_KEY2") # Gemma API Key
|
11 |
-
API_KEY_DEEPSEEK1 = os.getenv("OPENROUTER_API_KEY3") # DeepSeek First Query
|
12 |
-
API_KEY_DEEPSEEK2 = os.getenv("OPENROUTER_API_KEY4") # DeepSeek Final Refinement
|
13 |
|
14 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
llama_client = OpenAI(base_url="https://openrouter.ai/api/v1", api_key=API_KEY_LLAMA)
|
16 |
gemma_client = OpenAI(base_url="https://openrouter.ai/api/v1", api_key=API_KEY_GEMMA)
|
17 |
deepseek_client1 = OpenAI(base_url="https://openrouter.ai/api/v1", api_key=API_KEY_DEEPSEEK1)
|
@@ -39,7 +51,7 @@ def query_gemma(user_input, results):
|
|
39 |
except Exception as e:
|
40 |
results["Gemma"] = f"Error: {str(e)}"
|
41 |
|
42 |
-
# Function to query DeepSeek
|
43 |
def query_deepseek_1(user_input, results):
|
44 |
try:
|
45 |
completion = deepseek_client1.chat.completions.create(
|
@@ -50,7 +62,7 @@ def query_deepseek_1(user_input, results):
|
|
50 |
except Exception as e:
|
51 |
results["DeepSeek1"] = f"Error: {str(e)}"
|
52 |
|
53 |
-
# Function to refine responses using DeepSeek
|
54 |
def refine_response(user_input):
|
55 |
try:
|
56 |
results = {}
|
|
|
5 |
from dotenv import load_dotenv
|
6 |
|
7 |
# Load API keys from .env file
|
8 |
+
load_dotenv(override=True)
|
|
|
|
|
|
|
|
|
9 |
|
10 |
+
# Retrieve API keys (Default to "MISSING_KEY" for debugging)
|
11 |
+
API_KEY_LLAMA = os.getenv("OPENROUTER_API_KEY1", "MISSING_KEY")
|
12 |
+
API_KEY_GEMMA = os.getenv("OPENROUTER_API_KEY2", "MISSING_KEY")
|
13 |
+
API_KEY_DEEPSEEK1 = os.getenv("OPENROUTER_API_KEY3", "MISSING_KEY")
|
14 |
+
API_KEY_DEEPSEEK2 = os.getenv("OPENROUTER_API_KEY4", "MISSING_KEY")
|
15 |
+
|
16 |
+
# Debugging: Check API key values
|
17 |
+
print(f"Llama API Key: {API_KEY_LLAMA[:5]}...") # Show only first 5 characters
|
18 |
+
print(f"Gemma API Key: {API_KEY_GEMMA[:5]}...")
|
19 |
+
print(f"DeepSeek API Key 1: {API_KEY_DEEPSEEK1[:5]}...")
|
20 |
+
print(f"DeepSeek API Key 2: {API_KEY_DEEPSEEK2[:5]}...")
|
21 |
+
|
22 |
+
# Ensure all API keys are loaded
|
23 |
+
if "MISSING_KEY" in [API_KEY_LLAMA, API_KEY_GEMMA, API_KEY_DEEPSEEK1, API_KEY_DEEPSEEK2]:
|
24 |
+
raise ValueError("❌ ERROR: One or more API keys are missing from the .env file!")
|
25 |
+
|
26 |
+
# Create OpenAI clients for each model
|
27 |
llama_client = OpenAI(base_url="https://openrouter.ai/api/v1", api_key=API_KEY_LLAMA)
|
28 |
gemma_client = OpenAI(base_url="https://openrouter.ai/api/v1", api_key=API_KEY_GEMMA)
|
29 |
deepseek_client1 = OpenAI(base_url="https://openrouter.ai/api/v1", api_key=API_KEY_DEEPSEEK1)
|
|
|
51 |
except Exception as e:
|
52 |
results["Gemma"] = f"Error: {str(e)}"
|
53 |
|
54 |
+
# Function to query DeepSeek (First Query)
|
55 |
def query_deepseek_1(user_input, results):
|
56 |
try:
|
57 |
completion = deepseek_client1.chat.completions.create(
|
|
|
62 |
except Exception as e:
|
63 |
results["DeepSeek1"] = f"Error: {str(e)}"
|
64 |
|
65 |
+
# Function to refine responses using DeepSeek (Final API)
|
66 |
def refine_response(user_input):
|
67 |
try:
|
68 |
results = {}
|