Upload folder using huggingface_hub

app.py

@@ -3,6 +3,7 @@ from openai import OpenAI
 from pypdf import PdfReader
 import gradio as gr
 import json
+import os
 import datetime
 from pathlib import Path
 
@@ -22,7 +23,12 @@ def save_usage(total_cost):
         }, f)
 
 load_dotenv(override=True)
-client = OpenAI()
+# client = OpenAI()
+
+client = OpenAI(
+    api_key=os.getenv('DEEPSEEK_API_KEY'),
+    base_url="https://api.deepseek.com/v1"  # Example endpoint; replace with the actual DeepSeek endpoint
+)
 # === CONFIGURABLE BUDGET LIMIT (USD) ===
 USAGE_LIMIT_PER_DAY = 0.1  # USD
 USAGE_FILE = Path("daily_usage.json")
@@ -61,7 +67,7 @@ def chat(message, history):
         return "Daily usage limit reached. Please try again tomorrow."
 
     messages = [{"role": "system", "content": system_prompt}] + history + [{"role": "user", "content": message}]
-    response = client.chat.completions.create(model="gpt-4o-mini", messages=messages)
+    response = client.chat.completions.create(model="deepseek-chat", messages=messages)
     usage = response.usage
     input_tokens = usage.prompt_tokens
     output_tokens = usage.completion_tokens