Athspi commited on
Commit
cacc649
·
verified ·
1 Parent(s): 0cc372f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -8
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
- # Create OpenAI Clients for each model
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 for additional context
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-R1 (Final API)
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 = {}