fix(app): pdf qna playground

jina-ai · Apr 24, 2023 · e3e3c05 · e3e3c05
1 parent 9be2856
commit e3e3c05
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/lcserve/__main__.py b/lcserve/__main__.py
@@ -375,5 +375,14 @@ def babyagi(verbose):
     play(verbose=verbose)
 
 
+@playground.command(help='Play with pdf qna on JCloud.')
+def pdf_qna():
+    sys.path.append(os.path.join(os.path.dirname(__file__), 'playground', 'pdf_qna'))
+    from streamlit.web import cli as strcli
+
+    sys.argv = ['streamlit', 'run', 'lcserve/playground/pdf_qna/playground.py']
+    sys.exit(strcli.main())
+
+
 if __name__ == "__main__":
     serve()
diff --git a/lcserve/playground/pdf_qna/playground.py b/lcserve/playground/pdf_qna/playground.py
@@ -48,7 +48,7 @@ class Response(BaseModel):
 submit = st.button('Submit')
 
 
-def main():
+def play():
     if submit:
         if not openai_token:
             st.error('Please enter your OpenAI token')
@@ -85,4 +85,4 @@ def main():
 
 
 if __name__ == '__main__':
-    main()
+    play()