Update app.py

app.py

@@ -39,54 +39,54 @@ from streamlit_pdf_viewer import pdf_viewer
 
 
 
-# def consume_llm_api(prompt):
-#     """
-#     Sends a prompt to the LLM API and processes the streamed response.
-#     """
-#     url = "https://8417-201-238-124-65.ngrok-free.app/api/llm-response"
-#     headers = {"Content-Type": "application/json"}
-#     payload = {"prompt": prompt}
-
-#     try:
-#         print("Sending prompt to the LLM API...")
-#         with requests.post(url, json=payload, headers=headers, stream=True) as response:
-#             response.raise_for_status()
-#             print("Response from LLM API:\n")
-#             for line in response:
-#                 yield(line.decode('utf-8'))
-#             # print(type(response))
-#             # yield(response)
-#     except requests.RequestException as e:
-#         print(f"Error consuming API: {e}")
-#     except Exception as e:
-#         print(f"Unexpected error: {e}")
 def consume_llm_api(prompt):
+    """
+    Sends a prompt to the LLM API and processes the streamed response.
+    """
+    url = "https://8417-201-238-124-65.ngrok-free.app/api/llm-response"
+    headers = {"Content-Type": "application/json"}
+    payload = {"prompt": prompt}
+
+    try:
+        print("Sending prompt to the LLM API...")
+        with requests.post(url, json=payload, headers=headers, stream=True) as response:
+            response.raise_for_status()
+            print("Response from LLM API:\n")
+            for line in response:
+                yield(line.decode('utf-8'))
+            # print(type(response))
+            # yield(response)
+    except requests.RequestException as e:
+        print(f"Error consuming API: {e}")
+    except Exception as e:
+        print(f"Unexpected error: {e}")
+# def consume_llm_api(prompt):
     
-    client = Groq(
-        api_key="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"
-    )
+#     client = Groq(
+#         api_key="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"
+#     )
 
-    completion = client.chat.completions.create(
+#     completion = client.chat.completions.create(
         
-        model="llama-3.3-70b-versatile",
-        messages=[
-
-            {
-                "role": "system",
-                "content": prompt
-            },
-        ],
-
-        temperature=1,
-        # max_completion_tokens=1024,
-        top_p=1,
-        stream=True,
-        stop=None,
-    )
-
-    for chunk in completion:
-        if chunk.choices[0].delta.content:
-            yield chunk.choices[0].delta.content
+#         model="llama-3.3-70b-versatile",
+#         messages=[
+
+#             {
+#                 "role": "system",
+#                 "content": prompt
+#             },
+#         ],
+
+#         temperature=1,
+#         # max_completion_tokens=1024,
+#         top_p=1,
+#         stream=True,
+#         stop=None,
+#     )
+
+#     for chunk in completion:
+#         if chunk.choices[0].delta.content:
+#             yield chunk.choices[0].delta.content
 @st.cache_resource
 def encoding_model():
     """
@@ -111,37 +111,9 @@ def dataframe_info(data):
     value= data[:5]
     return str(value)
 
-# def extract_python_code(text):
-#     """
-#     Extracts code blocks from a given text.
-#     Supports triple-backtick blocks, indented code, and inline code.
-#     """
-#     code_snippets = []
-
-#     # Extract triple-backtick code blocks
-#     triple_backtick_blocks = re.findall(r"```(?:[\w+\-]*)\n(.*?)```", text, re.DOTALL)
-#     code_snippets.extend(triple_backtick_blocks)
-
-#     # Extract indented code blocks (4 spaces or tab)
-#     indented_blocks = re.findall(r"(?:^|\n)((?:    |\t).+(\n(?:    |\t).+)*)", text)
-#     code_snippets.extend([block[0] for block in indented_blocks])
-
-#     # Extract inline code snippets with single backticks
-#     inline_code = re.findall(r"`([^`\n]+)`", text)
-#     code_snippets.extend(inline_code)
-
-#     return [snippet.strip() for snippet in code_snippets if snippet.strip()]
 def extract_python_code(text):
-    # Match triple backtick Python code blocks ```python ... ```
-    code_blocks = re.findall(r"```(?:python)?\n(.*?)```", text, re.DOTALL)
-    
-    # If no triple-backtick code found, try to match indented blocks (e.g. 4 spaces)
-    if not code_blocks:
-        code_blocks = re.findall(r"(?:^|\n)(    .+?)(?=\n\S|\Z)", text, re.DOTALL)
-
-    # Clean indentation if needed
-    code_blocks = [re.sub(r"^\s{4}", "", block, flags=re.MULTILINE) for block in code_blocks]
-    return code_blocks
+    code_block=text.split("```")
+    return [code_block[1]]
 
 # @st.cache_resource
 def run_code_blocks(code_blocks,df):