Upload 7 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+storage/vector_store.json filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,12 +1,13 @@
 ---
-title: LLM Exp
-emoji: 🐨
-colorFrom: purple
-colorTo: blue
+title: Pharm
+emoji: 👁
+colorFrom: pink
+colorTo: indigo
 sdk: gradio
 sdk_version: 3.35.2
 app_file: app.py
 pinned: false
+duplicated_from: ssm123ssm/pharm
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,38 @@
+import os
+import openai
+from llama_index import StorageContext, load_index_from_storage
+import logging
+import sys
+
+logging.basicConfig(stream=sys.stdout, level=logging.INFO)
+logging.getLogger().handlers = []
+logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
+
+os.environ['OPENAI_API_KEY'] = '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'
+openai.api_key = os.environ['OPENAI_API_KEY']
+
+from llama_index import load_indices_from_storage
+storage_context = StorageContext.from_defaults(persist_dir='./storage')
+indx = load_indices_from_storage(storage_context)
+
+from llama_index.query_engine import CitationQueryEngine
+query_engine = CitationQueryEngine.from_args(index=indx[1], similarity_top_k=4)
+list_query_engine = indx[0].as_query_engine(
+    response_mode='tree_summarize',
+    use_async=True,
+)
+vector_query_engine = indx[1].as_query_engine(similarity_top_k=4)
+
+import gradio as gr
+def chatbot(input_text, is_qen):
+    engine = query_engine if is_qen else vector_query_engine
+    response = engine.query(input_text)
+    return response
+
+iface = gr.Interface(fn=chatbot,
+                     inputs=[gr.inputs.Textbox(lines=7, default="What are examples for short, intermediate and long acting glucocorticoids?", label="Enter your text here"), gr.Checkbox(label="Use Citation Engine instead of Query Engine.")],
+                     outputs="text",
+                     title="Project Rang and Dale")
+iface.launch()
+
+#%%

requirements.txt

@@ -0,0 +1,4 @@
+llama_index
+gradio~=3.35.2
+pypdf
+openai~=0.27.8

storage/graph_store.json

@@ -0,0 +1 @@
+{"graph_dict": {}}

storage/vector_store.json

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:faa5fcb34a27825e5c3a5f58813876a51f511b78668623b9fc98cd27e2251e35
+size 59050799