first attempt to hf spaces

README.md

@@ -5,7 +5,7 @@ colorFrom: indigo
 colorTo: gray
 sdk: gradio
 sdk_version: 3.45.2
-app_file: app.py
+app_file: gradio.py
 pinned: false
 ---
 

gradio.py

@@ -0,0 +1,17 @@
+import gradio as gr
+from hydra import compose, initialize
+from omegaconf import OmegaConf
+
+from .src.demo import App
+
+def main():
+    with initialize(version_base=None, config_path="conf", job_name="test_app"):
+        cfg = compose(config_name="config", overrides=["db=mysql", "db.user=me"])
+        
+        app = App(cfg)
+
+        webapp = gr.ChatInterface(fn=app.ask_chat, examples=["hello", "hola", "merhaba"], title="LLM4SciLit")
+        webapp.launch(share=True)
+
+if __name__ == "__main__":
+    main()

src/demo.py

@@ -94,6 +94,28 @@ class App:
         self.qa_model.initialize()
         print("Ready to answer your questions 🔥🔥\n")
 
+
+    ##################################################################################################
+    # App functionalities
+
+    def ask_paper(self, line):
+        """Ask a question about a paper."""
+        paper, line = shlex.split(line)
+        filter_dict = {"paper_title": paper}
+        print(f"\nLLM4SciLit: {self.qa_model.answer_question(line, filter_dict)['result']}\n")
+
+    def ask(self, line):
+        # print(f"\nLLM4SciLit: a bunch of nonsense\n")
+        print(f"\nLLM4SciLit: {self.qa_model.answer_question(line, {})['result']}\n")
+    
+    def ask_chat(self, line, history):
+        # print(f"\nLLM4SciLit: a bunch of nonsense\n")
+        return self.qa_model.answer_question(line, {})['result']
+
+
+    ##################################################################################################
+    # App modes        
+
     def run_interactive(self) -> None:
         self._bootstrap()
         cli = CLIApp(self, self.cfg)