feat(gradio): enhance gradio interface

examples/visualize_fallback_code.py

@@ -28,7 +28,7 @@ async def main():
 
     user_collection.set_fallback(fallback_collection).set_fallback(second_fallback_collection)
 
-    gradio_interface = await create_gradio_interface(user_collection=user_collection)
+    gradio_interface = create_gradio_interface(user_collection)
     gradio_interface.launch()
 
 

examples/visualize_views_code.py

@@ -11,14 +11,17 @@
 from dbally.llms.litellm import LiteLLM
 
 
-async def main():
+async def main() -> None:
     await country_similarity.update()
+
     llm = LiteLLM(model_name="gpt-3.5-turbo")
     dbally.event_handlers = [CLIEventHandler(), BufferEventHandler()]
+
     collection = dbally.create_collection("candidates", llm)
     collection.add(CandidateView, lambda: CandidateView(engine))
     collection.add(SampleText2SQLViewCyphers, lambda: SampleText2SQLViewCyphers(create_freeform_memory_engine()))
-    gradio_interface = await create_gradio_interface(user_collection=collection)
+
+    gradio_interface = create_gradio_interface(collection)
     gradio_interface.launch()
 
 

src/dbally/gradio/__init__.py

@@ -1,3 +1,3 @@
-from dbally.gradio.gradio_interface import create_gradio_interface
+from dbally.gradio.interface import create_gradio_interface
 
 __all__ = ["create_gradio_interface"]