Spaces:

pikaduck
/

policy-analyser

Running

Sakshi commited on 25 days ago

Commit

f74e0b9

1 Parent(s): 50a0334

pitching full suggestion

Files changed (1) hide show

app.py CHANGED Viewed

@@ -107,7 +107,8 @@ def main():
                     suggestion = next(
                         (item for item in response if item.get("stage") == "SUGGEST"), None
                     )['response']
-                    suggestion = suggestion.split('<POLICY_PITCH>')[-1].split('</POLICY_PITCH>')[0]
                     text = next(
                         (item for item in response if item.get("stage") == "OCR"), None
                     )['response']

                     suggestion = next(
                         (item for item in response if item.get("stage") == "SUGGEST"), None
                     )['response']
+                    # suggestion = suggestion.split('<POLICY_PITCH>')[-1].split('</POLICY_PITCH>')[0]
+                    suggestion = suggestion.replace('<POLICY_PITCH>', '').replace('</POLICY_PITCH>')
                     text = next(
                         (item for item in response if item.get("stage") == "OCR"), None
                     )['response']