Upload folder using huggingface_hub

test.py

@@ -1,96 +1,54 @@
 import gradio as gr
-
-def create_policy_popup_app():
-    # Your Google Drive file ID
-    pdf_id = "1WZCQpIWfelbxgBr8dNiW2rqVGaDyf-Gi"
-    embed_url = f"https://drive.google.com/file/d/{pdf_id}/preview"
-
-    custom_css = """
-    .pdf-modal {
-        display: none;
-        position: fixed;
-        z-index: 1000;
-        left: 0;
-        top: 0;
-        width: 100%;
-        height: 100%;
-        background-color: rgba(0,0,0,0.7);
-        overflow: auto;
-    }
-
-    .pdf-modal-content {
-        background-color: white;
-        margin: 2% auto;
-        padding: 20px;
-        width: 90%;
-        max-width: 900px;
-        height: 90%;
-        box-shadow: 0 5px 15px rgba(0,0,0,0.3);
-        border-radius: 8px;
-        position: relative;
-    }
-
-    .close-btn {
-        position: absolute;
-        right: 15px;
-        top: 15px;
-        font-size: 24px;
-        cursor: pointer;
-        background: #222;
-        color: white;
-        border: none;
-        border-radius: 4px;
-        padding: 5px 15px;
-        z-index: 1001;
-    }
-
-    .pdf-container {
-        width: 100%;
-        height: calc(100% - 40px);
-        margin-top: 40px;
-    }
-
-    .pdf-embed {
-        width: 100%;
-        height: 100%;
-        border: 1px solid #ddd;
-    }
-    """
-
-    modal_html = f"""
-    <div id="pdfModal" class="pdf-modal">
-        <div class="pdf-modal-content">
-            <button class="close-btn" onclick="document.getElementById('pdfModal').style.display='none';document.body.style.overflow='auto';">×</button>
-            <div class="pdf-container">
-                <iframe class="pdf-embed" src="{embed_url}" type="application/pdf"></iframe>
-            </div>
-        </div>
-    </div>
-    """
-
-    with gr.Blocks(css=custom_css) as app:
-        gr.Markdown("## View Hate Speech Policy")
-        gr.HTML(modal_html)
-
-        open_button = gr.Button("📄 Open Policy Document")
-
-        open_button.click(
-            fn=lambda: None,
-            inputs=None,
-            outputs=None,
-            js="""
-                () => {
-                    const modal = document.getElementById('pdfModal');
-                    if (modal) {
-                        modal.style.display = 'block';
-                        document.body.style.overflow = 'hidden';
-                    }
-                }
-            """
+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")
+
+def read_pdf(file):
+    if file is None:
+        return ""
+    reader = PyPDF2.PdfReader(file)
+    text = ""
+    for page in reader.pages:
+        text += page.extract_text()
+    return text
+
+def chat_with_openai(user_input, model_name, uploaded_pdf):
+    pdf_text = read_pdf(uploaded_pdf) if uploaded_pdf else ""
+    prompt = f"{pdf_text}\n\nUser Query: {user_input}" if pdf_text else user_input
+
+    try:
+        response = openai.ChatCompletion.create(
+            model=model_name,
+            messages=[{"role": "user", "content": prompt}]
         )
-
-    return app
-
-if __name__ == "__main__":
-    app = create_policy_popup_app()
-    app.launch()
+        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)
+
+    submit_btn.click(
+        fn=chat_with_openai,
+        inputs=[user_input, model_selector, uploaded_pdf],
+        outputs=response_output
+    )
+
+app.launch()