Update app.py

app.py

@@ -87,7 +87,6 @@ def process_pdf(file):
         ids=[file.name]  # Use the filename as the unique ID
     )
     
-
     return f"PDF wurde erfolgreich in ChromaDB gespeichert."
 
 def search_similar_documents(prompt):
@@ -97,7 +96,7 @@ def search_similar_documents(prompt):
     # Führe die Ähnlichkeitssuche durch
     results = collection.query(
         query_embeddings=[query_embedding],
-        n_results=20
+        n_results=3
     )
     
     # Formatiere die Ergebnisse
@@ -105,6 +104,56 @@ def search_similar_documents(prompt):
     for i, doc in enumerate(results["documents"][0]):
         metadata = results["metadatas"][0][i]
         filename = metadata["filename"]
-        formatted_results.append(f"### Dokument {i+1} (Dateiname: {filename})\n{doc}\n")
-    
-    return gr.Markdown(''.join(formatted_results))
+        formatted_results.append(f"{doc}\n")
+
+    ergebnis = f"{''.join(formatted_results)}"
+    ergebnis = gr.Markdown(ergebnis)
+    return ergebnis
+
+
+  with gr.Blocks() as chat:
+    gr.Markdown("### Ask the RKI Files", elem_classes="tab-header")
+    with gr.Row():       
+        llm_output = gr.Textbox(label="LLM Answer")
+    with gr.Row():
+        llm_prompt_input = gr.Textbox(label="Frage an das LLM", placeholder="Gib eine Frage ein")
+        llm_submit_button = gr.Button("send")  
+    llm_submit_button.click(ask_llm, inputs=llm_prompt_input, outputs=llm_output)
+
+with gr.Blocks() as upload:
+    gr.Markdown("### File upload", elem_classes="tab-header")
+    with gr.Row():
+        file_input = gr.File(label="Wähle eine PDF-Datei aus", type="filepath")
+        upload_output = gr.Textbox(label="Upload Status")
+    with gr.Row():
+        submit_button = gr.Button("upload")
+    submit_button.click(process_pdf, inputs=file_input, outputs=upload_output)
+
+with gr.Blocks() as suche:
+    gr.Markdown("### Datenbank durchsuchen", elem_classes="tab-header")
+    with gr.Row():
+        prompt_input = gr.Textbox(label="Suche nach ähnlichen Dokumenten", placeholder="Gib einen Suchbegriff ein")        
+    with gr.Row():
+        search_output = gr.Textbox(label="Ähnliche Dokumente")
+    with gr.Row():
+        search_button = gr.Button("Suchen")
+    search_button.click(search_similar_documents, inputs=prompt_input, outputs=search_output)
+
+#optional, Spracheingabe
+with gr.Blocks() as speech:
+    gr.Markdown("### Highspeed Voicebot", elem_classes="tab-header")
+    with gr.Row():
+        sr_outputs = gr.Textbox(label="Antwort")
+    with gr.Row():
+        sr_inputs = gr.Microphone(type="filepath") 
+    sr_inputs.change(transcribe_audio, inputs=sr_inputs, outputs=sr_outputs)
+
+# Erstelle die Gradio-Schnittstelle
+with gr.Blocks() as demo:
+    gr.TabbedInterface(
+        [chat, upload, suche],
+        ["Chat", "Upload", "Suche"]
+    )
+
+# Starte die Gradio-Anwendung
+demo.launch()