Create app.py

app.py

@@ -0,0 +1,29 @@
+import gradio as gr
+from autollm import AutoQueryEngine
+import os
+
+def query_engine(api_key, llm_model, documents):
+    api_key = os.getenv("HUGGINGFACE_API_KEY")
+    llm_api_base = "https://api-inference.huggingface.co/models/"
+
+    query_engine = AutoQueryEngine.from_defaults(
+        documents=documents,
+        llm_model=llm_model,
+        llm_api_base=llm_api_base,
+    )
+
+    return query_engine
+
+interface = gr.Blocks()
+
+with interface:
+    gr.Markdown("# AutoQueryEngine Interface")
+    llm_model = gr.Textbox(label="LLM Model", value="mistralai/Mixtral-8x7B-Instruct-v0.1")
+    documents = gr.Textbox(label="Documents (comma-separated paths or URLs)")
+    query = gr.Textbox(label="Query")
+    output = gr.Textbox(label="Output")
+
+    query_btn = gr.Button("Query")
+    query_btn.click(fn=query_engine, inputs=[api_key, llm_model, documents], outputs=output)
+
+interface.launch()