Upload folder using huggingface_hub

test.py

@@ -1,10 +1,10 @@
 import gradio as gr
-import openai
+from openai import OpenAI
 import os
 import PyPDF2
 
-# Read the API key from environment variable (you'll store it on Hugging Face Spaces)
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# Initialize OpenAI client with API key from environment variable
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
 def read_pdf(file):
     if file is None:
@@ -20,28 +20,33 @@ def chat_with_openai(user_input, model_name, uploaded_pdf):
     prompt = f"{pdf_text}\n\nUser Query: {user_input}" if pdf_text else user_input
 
     try:
-        response = openai.ChatCompletion.create(
+        response = client.chat.completions.create(
             model=model_name,
             messages=[{"role": "user", "content": prompt}]
         )
-        return response['choices'][0]['message']['content']
+        return response.choices[0].message.content
     except Exception as e:
         return f"Error: {str(e)}"
 
 with gr.Blocks() as app:
     gr.Markdown("# 🔥 OpenAI Chat + PDF Analysis Tool")
+    
     with gr.Row():
         model_selector = gr.Dropdown(
             choices=["gpt-3.5-turbo", "gpt-4", "gpt-4-turbo"],
             label="Select OpenAI Model",
             value="gpt-3.5-turbo"
         )
+    
     with gr.Row():
         uploaded_pdf = gr.File(label="Upload a PDF (optional)", file_types=[".pdf"])
+    
     with gr.Row():
         user_input = gr.Textbox(label="Your Prompt", placeholder="Ask anything...")
+    
     with gr.Row():
         submit_btn = gr.Button("Submit")
+    
     with gr.Row():
         response_output = gr.Textbox(label="OpenAI Response", lines=10, interactive=True)