Update app.py

app.py

@@ -2,21 +2,33 @@ import streamlit as st
 from streamlit_markmap import markmap
 from langchain.llms import OpenAI
 from langchain.prompts import PromptTemplate
+import os
+from langchain_openai import ChatOpenAI
 
 
-# Sidebar: Eingabe des OpenAI API Keys
-st.sidebar.header("Einstellungen")
-api_key = st.sidebar.text_input("OpenAI API Key", type="password")
-
+def remove_markdown_fences(text: str) -> str:
+    # Entferne führende Code-Fence (entweder "```markdown" oder "```")
+    if text.startswith("```markdown"):
+        text = text[len("```markdown"):].strip()
+    elif text.startswith("```"):
+        text = text[len("```"):].strip()
+    # Entferne abschließende Code-Fence "```", falls vorhanden
+    if text.endswith("```"):
+        text = text[:-3].strip()
+    return text
 
 # Streamlit-Konfiguration
 st.set_page_config(page_title="Mindmap Generator", layout="wide")
 st.title("Mindmap Ideen-Generator mit LangChain und OpenAI")
 
+# Sidebar: Eingabe des OpenAI API Keys
+st.sidebar.header("Einstellungen")
+api_key = st.sidebar.text_input("OpenAI API Key", type="password")
+os.environ['OPENAI_API_KEY'] = api_key
+
 # Texteingabe für den Brainstorming-Prompt
-user_prompt = st.text_area(
-    "Thema",
-    "Gib hier ein Thema ein, zu dem aus verschiedenen Perspektiven Ideen generiert werden sollen..."
+user_prompt = st.text_input(
+    "Thema"
 )
 
 # Button, um die Generierung zu starten
@@ -33,7 +45,8 @@ if st.button("Generiere Mindmap"):
             Generiere aus fünf Perspektiven mindestens jeweils zehn Ideen.
             
             Erstelle bitte eine Mindmap im Markdown-Format (Markmap), in der die Hauptideen und Unterideen hierarchisch strukturiert sind.
-            Nutze folgendes Format und gib nur das RAW aus:
+            
+            Nutze folgendes Format:
             # Hauptidee
             ## Unteridee 1
             ## Unteridee 2
@@ -42,16 +55,23 @@ if st.button("Generiere Mindmap"):
             prompt = PromptTemplate(template=template, input_variables=["topic"])
             full_prompt = prompt.format(topic=user_prompt)
             
-            # OpenAI LLM initialisieren (ggf. Umgebungsvariablen wie OPENAI_API_KEY setzen)
-            llm = OpenAI(model_name="gpt-4o", temperature=0.7, max_tokens=50000, openai_api_key=api_key)
-
+            llm = ChatOpenAI(
+                model="gpt-4o",
+                temperature=0,
+                max_tokens=None,
+                timeout=None,
+                max_retries=2
+            )
             
+            response = llm.invoke(full_prompt)
             # Anfrage an das LLM senden
-            mindmap_markdown = llm(full_prompt)
-            
+            mindmap_markdown = response.content
+            mindmap_markdown = remove_markdown_fences(mindmap_markdown)
+
             # Ausgabe der generierten Mindmap
-            st.markdown("## Generierte Mindmap (Markdown):")
-            st.code(mindmap_markdown, language="markdown")
+            with st.expander("See explanation"):
+                st.markdown("## Generierte Mindmap (Markdown):")
+                st.code(mindmap_markdown, language="markdown")
             
             # Darstellung der Mindmap mit streamlit_markmap
-            markmap(mindmap_markdown, height=600)
+            markmap(mindmap_markdown, height=800)