Upload folder using huggingface_hub

.env

@@ -0,0 +1,6 @@
+PALM="AIzaSyBVojf3nBKO_UITOwZtDVyAejW_2Qne1KY"
+HOST= "pod1.sonity.net"
+PORT= "8564"
+USER= "sonity"
+PASSWORD= "Docs4Sonity!"
+DBNAME= "postgres"

README.md

@@ -1,12 +1,6 @@
 ---
-title: Sonity Companies
-emoji: 🦀
-colorFrom: red
-colorTo: pink
+title: sonity_companies
+app_file: demo.py
 sdk: gradio
-sdk_version: 4.9.0
-app_file: app.py
-pinned: false
+sdk_version: 4.1.2
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

demo.py

@@ -0,0 +1,89 @@
+import numpy as np
+import pandas as pd
+import time
+from tqdm import tqdm
+import google.generativeai as palm
+import pandas as pd
+from langchain.chains import LLMChain
+
+
+from langchain.prompts import PromptTemplate
+
+import os
+
+import gradio as gr
+import io
+
+from langchain.llms import GooglePalm
+import pandas as pd
+#from yolopandas import pd
+
+from langchain.embeddings import GooglePalmEmbeddings
+from langchain.memory import ConversationBufferMemory
+from dotenv import load_dotenv
+from langchain.chains import RetrievalQA
+
+load_dotenv()
+
+from langchain.vectorstores.pgvector import PGVector
+
+embeddings = GooglePalmEmbeddings(google_api_key=os.environ['PALM'])
+
+host= os.environ['HOST']
+port= os.environ['PORT']
+user= os.environ['USER']
+password= os.environ['PASSWORD']
+dbname= os.environ['DBAME']
+
+
+CONNECTION_STRING = f"postgresql+psycopg2://{user}:{password}@{host}:{port}/{dbname}?sslmode=disable"
+
+store = PGVector(
+    collection_name="coms_data",
+    connection_string=CONNECTION_STRING,
+    embedding_function=embeddings,
+)
+
+retriever = store.as_retriever(
+    search_kwargs={"k": 3}
+    )
+
+llm = GooglePalm(temperature=0, google_api_key=os.environ['PALM'])
+
+def answer(query):
+# New chain to return context and sources
+  qa_stuff_with_sources = RetrievalQA.from_chain_type(
+      llm=llm,
+      chain_type="stuff",
+      retriever=retriever,
+      return_source_documents=True,
+      verbose=True,
+      )
+
+# To run the query, we use a different syntax since we're returning more than just the response text
+  responses = qa_stuff_with_sources({"query": query})
+
+  source_documents = responses["source_documents"]
+  source_content = [doc.page_content for doc in source_documents]
+  source_metadata = [doc.metadata for doc in source_documents]
+
+  result = responses['result']
+  result += "\n\n"
+  result += "Sources used:"
+  for i in range(len(source_content)):
+
+    result += "\n\n"
+    result += source_metadata[i]['company_name']
+    result += "\n\n"
+    result += source_metadata[i]['website']
+  return result
+
+
+demo = gr.Interface(
+
+    fn=answer,
+    inputs=["text"],
+    outputs=["text"],
+    title="Ask Sonity",
+)
+demo.launch(share=True)

requirements.txt

@@ -0,0 +1,10 @@
+google-ai-generativelanguage==0.1.0
+google-api-core==2.14.0
+google-auth==2.23.4
+google-generativeai==0.1.0rc1
+googleapis-common-protos==1.61.0
+gradio==4.7.1
+pgvector
+psycopg2-binary
+langchain==0.0.333
+python-dotenv==1.0.0