Update app.py

app.py

@@ -0,0 +1,88 @@
+import os
+import openai
+import pandas as pd
+import gradio as gr
+from openai import OpenAI
+from google.colab import userdata, drive
+from langchain_community.embeddings.sentence_transformer import SentenceTransformerEmbeddings
+from langchain_community.vectorstores import Chroma
+
+client=OpenAI(
+    api_key=os.getenv("OPENAI_API_KEY")
+)
+
+model_name = 'gpt-3.5-turbo'
+embedding_model = SentenceTransformerEmbeddings(model_name="thenlper/gte-large")
+persisted_vectordb_location = '/vector_db/'
+collection_name = 'companies-10K-2023'
+
+vectorstore_persisted = Chroma(
+    collection_name=collection_name,
+    embedding_function=embedding_model,
+    persist_directory=persisted_vectordb_location
+)
+
+source = set()
+document_names = set()
+
+for metadata in stored_documents['metadatas']:
+    # Extract the source and use os.path.basename to get only the file name
+    source = metadata.get('source', 'No source found')
+    document_names.add(os.path.basename(source))
+
+document_list = list(document_names)
+document_list.insert(0, "All Documents")
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column(scale=1):
+            
+            document_dropdown = gr.Dropdown(
+                choices=document_list,
+                label="Document",                
+            )
+            
+            
+            temperature_slider = gr.Slider(
+                minimum=0, 
+                maximum=1, 
+                step=0.1, 
+                #value=0.3,  
+                label="Temperature", 
+                info="Controls randomness: 0 = deterministic, 1 = creative/unexpected answers. If you can't get an answer try increasing the temperature but keep in mind that the accuracy can lower by doing this."
+            )
+            
+            
+            question_input = gr.Textbox(
+                label="Enter your question", 
+                placeholder="Type your question here..."
+            )
+            
+            
+            analyze_button = gr.Button("Analyze and Answer")
+        
+        with gr.Column(scale=1):
+            status_button = gr.Button(value="Ready", interactive=False, elem_id="status_button")
+            
+            # Read-only textbox for Question Analysis
+            question_analysis_output = gr.Textbox(
+                label="Question Analysis", 
+                placeholder="The analysis will be shown here...",
+                interactive=False
+            )
+
+    with gr.Row():
+        final_answer_output = gr.Textbox(
+            label="Final Answer", 
+            placeholder="The final answer will be shown here...",
+            interactive=False
+            )
+    
+    # Button click action to call the analyze_and_answer function
+    #analyze_button.click(
+    #    analyze_and_answer, 
+    #    inputs=[document_dropdown, temperature_slider, question_input], 
+    #    outputs=[question_analysis_output, final_answer_output]
+    #)
+
+demo.launch()