andrewxchen commited on
Commit
0215a4d
·
verified ·
1 Parent(s): 5d3cb1a

Add excerpt conversion functionality

Browse files
Files changed (1) hide show
  1. app.py +16 -1
app.py CHANGED
@@ -267,6 +267,16 @@ Return results in JSON format, e.g., {"GI":1, "Persuade":0, ...}
267
 
268
  return summary
269
 
 
 
 
 
 
 
 
 
 
 
270
  @staticmethod
271
  def load_prompt(filename):
272
  """Load prompt from file"""
@@ -362,9 +372,14 @@ def main():
362
 
363
  file_type = st.radio(
364
  "Select input type:",
365
- ["Audio/Video Recording", "Text Transcript"]
366
  )
367
 
 
 
 
 
 
368
  if file_type == "Audio/Video Recording":
369
  supported_formats = (
370
  st.session_state.media_processor.supported_formats['audio'] +
 
267
 
268
  return summary
269
 
270
+ def convert_excerpt(text):
271
+ """Convert text excerpt input into revised behavioral code"""
272
+ model = genai.GenerativeModel('models/gemini-2.0-flash-thinking-exp')
273
+ prompt = self.load_prompt(f"prompts/prompts/miti_convert.md")
274
+ response = model.generate_content(
275
+ prompt + '\n' + text ,
276
+ generation_config=generation_config
277
+ )
278
+ return response.text
279
+
280
  @staticmethod
281
  def load_prompt(filename):
282
  """Load prompt from file"""
 
372
 
373
  file_type = st.radio(
374
  "Select input type:",
375
+ ["Audio/Video Recording", "Text Transcript", "Text Excerpt Conversion"]
376
  )
377
 
378
+ if file_type == "Text Excerpt Conversion":
379
+ txt = st.text_area("Excerpt to analyze")
380
+ if st.button("Run"):
381
+ st.markdown(st.session_state.miti_analyzer.convert_excerpt(txt))
382
+
383
  if file_type == "Audio/Video Recording":
384
  supported_formats = (
385
  st.session_state.media_processor.supported_formats['audio'] +