remove: js

app.py

@@ -200,12 +200,17 @@ class TranscriptProcessor:
                 print("No agenda found in any description tags.")
                 return
 
+            print(self.speaker_mapping)
+
             prompt = (
-                f"Given the speaker mapping {self.speaker_mapping}, agenda:\n{agenda}, and the transcript: {self.formatted_transcript}\n\n"
+                f"Given the original speaker mapping {self.speaker_mapping}, agenda:\n{agenda}, and the transcript: {self.formatted_transcript}\n\n"
                 "Some speaker names in the mapping might have spelling errors or be incomplete."
+                "Remember that the content in agenda is accurate and transcript can have errors so prioritize the spellings and names in the agenda content."
+                "If the speaker name and introduction is similar to the agenda, update the speaker name in the mapping."
                 "Please correct the names based on the agenda. Return the corrected mapping in JSON format as "
                 "{'spk_0': 'Correct Name', 'spk_1': 'Correct Name', ...}."
                 "You should only update the name if the name sounds very similar, or there is a good spelling overlap/ The Speaker Introduction matches the description of the Talk from Agends. If the name is totally unrelated, keep the original name."
+                "You should always include all the speakers in the mapping from the original mapping, even if you don't update their names. i.e if there are 4 speakers in original mapping, new mapping should have 4 speakers always, ignore all the other spekaers in the agenda. I REPEAT DO NOT ADD OTHER NEW SPEAKERS IN THE MAPPING."
             )
 
             client = OpenAI()
@@ -254,7 +259,7 @@ class TranscriptProcessor:
             self.formatted_transcript = "\n".join(formatted_segments)
 
         except requests.exceptions.RequestException as e:
-            print(f"Error fetching agenda from URL: {str(e)}")
+            print(f"  ching agenda from URL: {str(e)}")
         except Exception as e:
             print(f"Error correcting speaker mapping: {str(e)}")
 
@@ -447,12 +452,34 @@ In the URL, make sure that after RSID there is ? and then rest of the fields are
         )
 
         response = completion.choices[0].message
-        if response.function_call:
-            args = json.loads(response.function_call.arguments)
+        messages.append(response)
+
+        if response.tool_calls:
+            args = eval(response.tool_calls[0].function.arguments)
             url = args.get("url", None)
             if url:
                 transcript_processor.correct_speaker_mapping_with_agenda(url)
-                return "Speaker names corrected based on the agenda."
+                corrected_speaker_mapping = transcript_processor.speaker_mapping
+                function_call_result_message = {
+                    "role": "tool",
+                    "content": json.dumps(
+                        {
+                            "speaker_mapping": f"Corrected Speaker Mapping is: {corrected_speaker_mapping}\n, All speakers should be addressed via this mapping. The next message should be comparing old speaker names (do not use spk_0, spk_1, use the old names) and corrected speaker names.",
+                        }
+                    ),
+                    "name": response.tool_calls[0].function.name,
+                    "tool_call_id": response.tool_calls[0].id,
+                }
+
+                # messages.append(response.choices[0]["message"])
+                messages.append(function_call_result_message)
+                completion_payload = {"model": "gpt-4o-mini", "messages": messages}
+                # print("messages", messages[3])
+                response = client.chat.completions.create(**completion_payload)
+                # print("no error here")
+
+                return response.choices[0].message.content
+
             else:
                 return "No URL provided for correcting speaker names."
 
@@ -493,39 +520,40 @@ def create_chat_interface():
         flex-grow: 1 !important;
     }
     """
-    js = """
-function createIframeHandler() {
-    let iframe = document.getElementById('link-frame');
-    if (!iframe) {
-        iframe = document.createElement('iframe');
-        iframe.id = 'link-frame';
-        iframe.style.position = 'absolute';
-        iframe.style.width = '1px';
-        iframe.style.height = '1px';
-        iframe.style.right = '-100px';
-        iframe.style.bottom = '-100px';
-        iframe.style.display = 'none'; // Hidden initially
-        document.body.appendChild(iframe);
-    }
-
-    document.addEventListener('click', function (event) {
-        var link = event.target.closest('a');
-        if (link && link.href) {
-            try {
-                iframe.src = link.href;
-                iframe.style.display = 'block'; // Show iframe on link click
-                event.preventDefault();
-                console.log('Opening link in iframe:', link.href);
-            } catch (error) {
-                console.error('Failed to open link in iframe:', error);
-            }
-        }
-    });
-
-    return 'Iframe handler initialized';
-}
-"""
-    with gr.Blocks(fill_height=True, fill_width=True, css=css, js=js) as demo:
+    #     js = """
+    # function createIframeHandler() {
+    #     let iframe = document.getElementById('link-frame');
+    #     if (!iframe) {
+    #         iframe = document.createElement('iframe');
+    #         iframe.id = 'link-frame';
+    #         iframe.style.position = 'absolute';
+    #         iframe.style.width = '1px';
+    #         iframe.style.height = '1px';
+    #         iframe.style.right = '-100px';
+    #         iframe.style.bottom = '-100px';
+    #         iframe.style.display = 'none'; // Hidden initially
+    #         document.body.appendChild(iframe);
+    #     }
+
+    #     document.addEventListener('click', function (event) {
+    #         var link = event.target.closest('a');
+    #         if (link && link.href) {
+    #             try {
+    #                 iframe.src = link.href;
+    #                 iframe.style.display = 'block'; // Show iframe on link click
+    #                 event.preventDefault();
+    #                 console.log('Opening link in iframe:', link.href);
+    #             } catch (error) {
+    #                 console.error('Failed to open link in iframe:', error);
+    #             }
+    #         }
+    #     });
+
+    #     return 'Iframe handler initialized';
+    # }
+    # """
+
+    with gr.Blocks(fill_height=True, fill_width=True, css=css) as demo:
         chatbot = gr.Chatbot(
             elem_id="chatbot_box",
             layout="bubble",