Upload app.py

app.py

@@ -25,9 +25,11 @@ def fetch_arxiv_summary(query, sort_by="relevance", sort_order="descending", max
         root = ET.fromstring(xml_data)
         summaries = []
         for entry in root.findall(".//{http://www.w3.org/2005/Atom}entry"):
+            title = entry.find("{http://www.w3.org/2005/Atom}title")
+            link = entry.find("{http://www.w3.org/2005/Atom}id")
             summary = entry.find("{http://www.w3.org/2005/Atom}summary")
-            if summary is not None:
-                summaries.append(summary.text.strip())
+            if summary is not None and title is not None and link is not None:
+                summaries.append(f"Titel: {title.text.strip()}\nLink: {link.text.strip()}\nZusammenfassung: {summary.text.strip()}")
         return summaries if summaries else ["Keine relevanten Studien gefunden."]
     except Exception as e:
         return [f"Fehler beim Abrufen der Studie: {str(e)}"]
@@ -50,7 +52,7 @@ def respond(
     study_info = "\n".join(study_summaries)
 
     # Nachrichten vorbereiten
-    messages = [{"role": "system", "content": system_message}]
+    messages = [{"role": "system", "content": f"{system_message}\nBitte gib immer die Links zu den Studien aus der Suche an."}]
     for val in history:
         if val[0]:
             messages.append({"role": "user", "content": val[0]})
@@ -73,6 +75,12 @@ def respond(
         yield response
 
 # Gradio-Interface mit zusätzlichen Eingaben
+def create_intro_text():
+    return ("Willkommen beim Chatbot! Dieser Chatbot verwendet KI, um Ihre Fragen zu beantworten und relevante Studien "
+            "aus der arXiv-Datenbank abzurufen. Geben Sie eine Frage ein, und der Bot liefert Ihnen basierend auf Ihrem "
+            "Suchbegriff Studien mit Titel, Link und Zusammenfassung. Zusätzlich können Sie die Sortierung und maximale "
+            "Anzahl der Ergebnisse anpassen.")
+
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
@@ -84,6 +92,7 @@ demo = gr.ChatInterface(
         gr.Dropdown(label="Sortierreihenfolge", choices=["ascending", "descending"], value="descending"),
         gr.Slider(label="Maximale Ergebnisse", minimum=1, maximum=50, value=20, step=1),
     ],
+    description=create_intro_text()
 )
 
 if __name__ == "__main__":